python - 从重复数组中绘制 python 绘图

Question

我的数据排列如下:

x = [0,1,2,0,1,2,0,1,2]
y = [0,0,0,1,1,1,2,2,2]
z = [2,4,3,3,5,1,1,2,1]

我如何绘制它以便获得 3x3 imshow 或等高线图？

Answer 1

如果 z 实际上是 x 和 y 的函数，并且您想要一个 countourf 对于绘图，您可以使用 numpy reshape 数组，然后绘制它:

import matplotlib.pyplot as plt
import numpy as np
x = np.array([0,1,2,0,1,2,0,1,2])
y = np.array([0,0,0,1,1,1,2,2,2])
z = np.array([2,4,3,3,5,1,1,2,1])
cols = np.unique(x).shape[0]
X = x.reshape(-1, cols)
Y = y.reshape(-1, cols)
Z = z.reshape(-1, cols)
plt.contourf(X,Y,Z)
plt.show()