我如何绘制一个正方形，并使用一个二维值数组对其进行颜色映射

Question

例如，我有两个x、y坐标的数组和一个对应于每个点的方形网格。

x = np.linspace(0,10,50)
y = np.linspace(-5,12,50)
square = np.empty((50,50))

我正试图画出一个由平方值构成的正方形曲面。我确实让它用于曲面绘制，但问题是曲面的形状不规则，我无法在上面绘制一些散射点，而不会出现可见性问题。

我想知道是否有一种方法可以让它作为二维图而不是三维图工作，因为Z cordinate可以用cmap代替，并且有意义。有什么建议吗？

Answer 1

所以我确实找到了一个解决方案：

x_mesh, y_mesh = np.meshgrid(x,y)
fig = plt.figure(dpi = 200)
plt.scatter(x_mesh,y_mesh, c = square, cmap = 'viridis')
plt.show

有了足够的点，它就有了一个好的表面。