Thứ tự mảng đường viền đầu vào matplotlib

Question

Câu hỏi: đường viền từ matplotlib mong đợi cho mảng 2D đầu vào là gì?

Xây dựng: Matplotlib tài liệu hướng dẫn đường viền nói rằng các cuộc gọi thông thường là

x_axis = np.linspace(10,100,n_x) 
y_axis = np.linspace(10,100,n_y) 
matplotlib.pyplot.contour(x_axis, y_axis, scalar_field) 

đâu scalar_field phải là một mảng 2D. Ví dụ, scalar_field có thể được tạo ra bởi

scalar_field = np.array([(x*y) for x in x_axis for y in y_axis]) 
scalar_field = scalar_field.reshape(n_x, n_y) 

Nếu scalar_field được trao cho đường viền,

plt.contour(x_axis, y_axis,scalar_field) #incorrect 

định hướng của cốt truyện là không chính xác (xoay). Để khôi phục đúng hướng, scalar_field phải được chuyển vị trí:

plt.contour(x_axis, y_axis,scalar_field.transpose()) #correct 

Vì vậy, thứ tự đường viền mong đợi rằng scalar_field có là gì?

Answer 1

Bạn nên vẽ bằng mảng contour qua cũng 2-D cho X và Y, sau đó mỗi điểm trong scalar_field mảng của bạn sẽ tương ứng với một phối hợp (x, y) trong X và Y. Bạn có thể thoải mái tạo X và Y sử dụng numpy.meshgrid:

import matplotlib.pyplot as plt 
import numpy as np 

X, Y = np.meshgrid(x_axis, y_axis, copy=False, indexing='xy') 
plt.contour(X, Y, scalar_field)  

Đối số indexing có thể được thay đổi để 'ij' nếu bạn muốn x phối hợp để đại diện cho dòng và y để đại diện cho cột, nhưng trong trường hợp này scalar_fied phải được tính bằng ij lập chỉ mục.

Answer 2

Các giá trị x dự kiến ​​sẽ tương ứng với cột dữ liệu, không phải là hàng (ví dụ: x là trục ngang và y là trục thẳng đứng). Bạn có nó đảo ngược, đó là lý do tại sao bạn phải chuyển đổi các giá trị z để làm cho nó hoạt động.

Để tránh yêu cầu chuyển vị, tạo mảng của bạn như:

scalar_field = np.array([(x*y) for y in y_axis for x in x_axis]) 
scalar_field = scalar_field.reshape(n_y, n_x)