3

我正在尝试创建一个等高线图，其中 x 坐标标记为 EF，y 标记为 EB，z 为标记为 a 的函数。它返回下面发布的长错误。任何帮助，将不胜感激。错误是

File "contour.py", line 19, in <module>
c = plt.contour(EF,EB,a)
  File "/usr/lib/pymodules/python2.7/matplotlib/pyplot.py", line 2016, in contour
    ret = ax.contour(*args, **kwargs)
  File "/usr/lib/pymodules/python2.7/matplotlib/axes.py", line 7326, in contour
    return mcontour.QuadContourSet(self, *args, **kwargs)
  File "/usr/lib/pymodules/python2.7/matplotlib/contour.py", line 1088, in __init__
    ContourSet.__init__(self, ax, *args, **kwargs)
  File "/usr/lib/pymodules/python2.7/matplotlib/contour.py", line 686, in __init__
    self._process_args(*args, **kwargs)
  File "/usr/lib/pymodules/python2.7/matplotlib/contour.py", line 1101, in _process_args
    x, y, z = self._contour_args(args, kwargs)
  File "/usr/lib/pymodules/python2.7/matplotlib/contour.py", line 1148, in _contour_args
    x,y,z = self._check_xyz(args[:3], kwargs)
  File "/usr/lib/pymodules/python2.7/matplotlib/contour.py", line 1180, in _check_xyz
    raise TypeError("Input z must be a 2D array.")
TypeError: Input z must be a 2D array.
4

2 回答 2

8

该错误指出

TypeError: Input z must be a 2D array.

如果您查看输入对象的大小：

print EF.shape, EB.shape, a.shape
(51,) (51,) (51,)

你会看到这些不是二维数组。您是否打算使用XandY来代替？

当我更改为

a = ((1+.5*(np.exp(1.7*X)+np.exp(1.7*Y)+np.exp(1.7*(X+Y))))/(1+np.exp(1.7*X)+np.exp(1.7*Y)+np.exp(1.7*(X+Y))))
c = plt.contour(EF,EB,a,30)

输出是

在此处输入图像描述

看起来你可能需要调整你的参数空间，因为所有有趣的东西都在(0,0).

于 2013-08-23T16:18:51.077 回答
4

你只需要创建a一个网格，使用XabdY而不是EFand EB：

a = ((1+.5*(np.exp(1.7*Y)+np.exp(1.7*X)+np.exp(1.7*(Y+X))))/(1+np.exp(1.7*Y)+np.exp(1.7*X)+np.exp(1.7*(Y+X))))

另一件事，如果您使用copy=False它创建网格网格可能会阻止您耗尽内存：

(X,Y) = np.meshgrid(EF,EB, copy=False)

在这种情况下，它会创建原始一维数组的视图。

于 2013-08-23T16:19:21.230 回答