如何将数组中的值赋给(x,y)点

时间:2014-03-17 15:56:48

标签: python arrays matplotlib

我有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).我理解错误,但不知道如何解决。

1 个答案:

答案 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)