我有3600 x值和3600 y值。和阵列(180x20)。我需要将其绘制为contourplot并将第一个x值和第一个y值分配给第一个数组值(A [0,0]),依此类推......我该怎么做?
我的想法:
x,y = np.meshgrid(x,y)
plt.contourf(x,y,A)
不起作用:TypeError: Shape of x does not match that of z: found (3600, 3600) instead of (180, 20).
我理解错误,但不知道如何解决。
答案 0 :(得分:3)
我认为你对meshgrid的做法有点困惑。听起来你只想要reshape
。
numpy.meshgrid
作为meshgrid的一个例子,我们假设我们有一个5x5网格" z"值:
import numpy as np
z = np.random.random((5,5))
我们有一行的x坐标和列的y坐标:
ny, nx = z.shape
x = np.linspace(5, 7, nx)
y = np.linspace(-2, 8, ny)
print '-----Z-----\n', z, '\n'
print '-----X-----\n', x, '\n'
print '-----Y-----\n', y, '\n'
所以,在这一点上,我们有:
-----Z-----
[[ 0.70319561 0.0141277 0.17580355 0.20411183 0.81714624]
[ 0.45093838 0.18241847 0.27477369 0.4881957 0.62157783]
[ 0.83172549 0.75278372 0.64856436 0.76651935 0.0152465 ]
[ 0.50908933 0.51557264 0.9975723 0.39579782 0.71333262]
[ 0.58998339 0.59205064 0.42716255 0.14138964 0.38212301]]
-----X-----
[ 5. 5.5 6. 6.5 7. ]
-----Y-----
[-2. 0.5 3. 5.5 8. ]
numpy.meshgrid
用于获取列和行坐标的一维数组,并将它们转换为与z
形状匹配的2D数组:
yy, xx = np.meshgrid(y, x)
print '-----XX----\n', xx, '\n'
print '-----YY----\n', yy, '\n'
这会产生:
-----XX----
[[ 5. 5. 5. 5. 5. ]
[ 5.5 5.5 5.5 5.5 5.5]
[ 6. 6. 6. 6. 6. ]
[ 6.5 6.5 6.5 6.5 6.5]
[ 7. 7. 7. 7. 7. ]]
-----YY----
[[-2. 0.5 3. 5.5 8. ]
[-2. 0.5 3. 5.5 8. ]
[-2. 0.5 3. 5.5 8. ]
[-2. 0.5 3. 5.5 8. ]
[-2. 0.5 3. 5.5 8. ]]
如果我正确理解你,你的x和y数组都是3600个元素列表,你的z数组是180x20。 (请注意180 * 20 == 3600
)因此,我认为您拥有的数据相当于:
yourx, youry = xx.flatten().tolist(), yy.flatten().tolist()
显然,您的数据要大得多,但如果我们扩展上面的示例,它会看起来像:
---yourx---
[5.0, 5.0, 5.0, 5.0, 5.0, 5.5, 5.5, 5.5, 5.5, 5.5, 6.0, 6.0, 6.0, 6.0, 6.0, 6.5, 6.5, 6.5, 6.5, 6.5, 7.0, 7.0, 7.0, 7.0, 7.0]
---youry---
[-2.0, 0.5, 3.0, 5.5, 8.0, -2.0, 0.5, 3.0, 5.5, 8.0, -2.0, 0.5, 3.0, 5.5, 8.0, -2.0, 0.5, 3.0, 5.5, 8.0, -2.0, 0.5, 3.0, 5.5, 8.0]
因此,您只想将列表重塑为与z
形状相同的2D数组。例如:
print np.reshape(yourx, z.shape)
产量
[[ 5. 5. 5. 5. 5. ]
[ 5.5 5.5 5.5 5.5 5.5]
[ 6. 6. 6. 6. 6. ]
[ 6.5 6.5 6.5 6.5 6.5]
[ 7. 7. 7. 7. 7. ]]
换句话说,你想要:
plt.contourf(np.reshape(yourx,z.shape),np.reshape(youry,z.shape),z)