如何使用3个数组进行彩色2D网格 [英] How to do colored 2D grid with 3 arrays
问题描述
我有三个相等长度的x,y和z数组. x和y数组是网格的x轴和y轴. z数组将确定网格块的颜色.例如,
I have three arrays of equal length x, y, and z. The x and y arrays are the x-axis and y-axis for the grid. The z array will determine the color of the the grid block. For example,
x = [10, 10, 10, 20, 20, 20, 30, 30, 30]
y = [10, 20, 30, 10, 20, 30, 10, 20, 30]
z = [100, 54, 32, 67, 71, 88, 100, 15, 29]
使用
ax.plot_trisurf(x, y, z, cmap=cm.RdYlGn)
或
ax.bar3d(x, y, [0] * len(x), 100, 100, z, cmap=cm.RdYlGn)
但是我正在寻找类似的东西 此
But I am looking for something like this
推荐答案
np.meshgrid
返回两个2D数组的元组,您可以直接解压缩
np.meshgrid
returns a tuple of two 2D arrays, which you can unpack directly
X,Y = np.meshgrid(x,y)
但是,您不需要那些用于imshow图.您需要的和代码中缺少的是z
值的2D数组.这将是提供给imshow
的数组.
However, you don't need to those for an imshow plot. What you need and what you lack in your code is the 2D array of z
values. This would be the array to provide to imshow
.
img = plt.imshow(Z)
如果要改用meshgrid,则可以使用X
和Y
值,
If you want to use meshgrid instead, you can use your X
and Y
values,
plt.pcolormesh(X,Y,Z)
查看示例数据,可以使用imshow:
Seeing the example data, you can use imshow:
x = [10, 10, 10, 20, 20, 20, 30, 30, 30]
y = [10, 20, 30, 10, 20, 30, 10, 20, 30]
z = [100, 54, 32, 67, 71, 88, 100, 15, 29]
import matplotlib.pyplot as plt
import numpy as np
z = np.array(z).reshape(3,3)
plt.imshow(z,extent=[5,35,5,35])
plt.show()
这篇关于如何使用3个数组进行彩色2D网格的文章就介绍到这了,希望我们推荐的答案对大家有所帮助,也希望大家多多支持IT屋!