将X、Y、Z值转换为矩阵图
将X、Y、Z值转换为矩阵图
提问于 2019-11-12 16:11:09
我有以下列表：
x_list = [7,10,3,9,9]
y_list = [2,5,1,3,2]
z_list = [4,1,5,2,3]对于每个x值和y值，都有一个z值。例如：x=7和y=2的z值为4。
我想将其转换为矩阵，如下所示：
…为了能够创建具有正确的x轴和y轴以及颜色条的矩阵图(此处缺少颜色条，最好将零值设置为黑色)：
这在Excel中相当简单，但我在python中却遇到了困难，因为我是matplotlib的新手。
编辑
我可能会使用plt.imshow()来绘制图
回答 1
Stack Overflow用户
发布于 2019-11-12 17:02:01
NumPy的unique和meshgrid函数是你在这里的朋友，他们可以根据你的坐标设置适当的网格。然后，您只需要在生成的X和Y中找到与您的坐标相对应的适当索引。最后，在一些与X和Y维度相同的全零Z矩阵中，可以将刚找到的索引处的元素设置为原始z数据。
下面是一些代码：
import numpy as np
x = np.array([7, 10, 3, 9, 9])
y = np.array([2, 5, 1, 3, 2])
z = np.array([4, 1, 5, 2, 3])
# Calculate X, Y grid from x and y, initialize Z data with zero
X, Y = np.meshgrid(np.unique(x), np.unique(y))
Z = np.zeros(X.shape)
# Find proper indices in grid, and set Z values accordingly
idx = np.squeeze([np.argwhere(np.logical_and(X == x[idx], Y == y[idx])) for idx in np.int32(np.arange(x.size))])
Z[idx[:, 0], idx[:, 1]] = z
print(Z)并且，输出是：
[[5. 0. 0. 0.]
[0. 4. 3. 0.]
[0. 0. 2. 0.]
[0. 0. 0. 1.]]关于绘图:我想不出Matplotlib函数来创建类似于所示的绘图。所以，也许可以给出一个额外的提示，你想要的是哪个函数。
希望这能有所帮助！
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/58814462
复制相关文章
相似问题
腾讯云开发者
