Meu objetivo é traçar um gráfico de contorno de fluxo, mas em vez de curvas de contorno fechadas, plt.contour()retorna uma série de linhas com a mesma altura.
Psi é definido como np.arraye tem um formato de 320 por 200.
fig, ax = plt.subplots()
r_end = grid_start[0] + grid_step[0] * grid_size[0]
z_end = grid_start[1] + grid_step[1] * grid_size[1]
X = np.arange(grid_start[0], r_end, grid_step[0]) # shape (200,)
Y = np.arange(grid_start[1], z_end, grid_step[1]) # shape (320,)
x, y = np.meshgrid(X, Y)
CS = ax.contour(x, y, Psi)
ax.clabel(CS, fontsize=10)
O resultado parece muito estranho:
Também criei gráficos de contorno com algum nivelamento para melhor visibilidade.
O resultado desejado é algo como isto: Contornos Psi desejados
Se eu remodelar o array fornecido para (3200,20), a periodicidade será perdida, mas ainda é estranho.
Como posso resolver esse problema? Obrigado pela ajuda.