Python：绘制等值线图时出现类型错误

Question

尽管使用了搜索功能，我还是无法找到答案。我有两个假设，但不知道它们可以应用到什么程度。现在的问题是：

我想绘制轮廓。为此，我有以下 python 代码：

import numpy as np
import matplotlib.pyplot as plt

xi=list_of_distance
yi=list_of_angle

x = np.arange(0,54,0.2)
y = np.arange(0,180,2.0)

Z = np.histogram2d(xi,yi,bins=(274,90))
X, Y = np.meshgrid(x, y)

plt.contour(X,Y,Z)
plt.ylabel('angles')
plt.xlabel('distance')
plt.colorbar()
plt.show()

xi 和 yi 是包含浮点值的列表。 x 和 y 定义“间隔”...例如： x 以 0.2 步长生成一个值从 0 到 54 的列表 y 在 2.0 步中生成一个值从 0 到 180 的列表，

ZI 使用 numpy 函数创建 2D 直方图。事实上，这似乎是造成麻烦的地方。

当调用函数 plt.contour(X,Y,Z) 时，出现以下错误消息：

...文件“/usr/lib/pymodules/python2.7/numpy/ma/core.py”，第 2641 行，new _data = np.array(数据, dtype=dtype, 复制=复制, subok=True, ndmin=ndmin) ValueError：使用序列设置数组元素。

现在假设可能导致此问题的原因：

1. 它似乎需要一个数组，但它接收的不是一个 numpy 数组，而是一个列表

，或者

2. 我们得到了比其他行短的行（我想到了这个想法，在一位同事跑完之后一年前就遇到过这样的问题 - 通过弄清楚最后一行比所有其他行短 2 个元素，已经解决了这个问题......）

Answer 1

正如 @rocksportrocker 所暗示的，您需要考虑 histogram2d 除了直方图之外还返回边缘。另一个细节是您可能希望显式传递一个范围，否则将根据数据中的实际最小值和最大值为您选择一个范围。然后，您需要将绘图的边缘转换为单元格中心。像这样的东西：

import numpy as np
import matplotlib.pyplot as plt

n = 1000000                     # how many data points
xmin, xmax = 0.0, 54.0          # distances
ymin, ymax = 0.0, 180.0         # angles

# make up some random data
xi=np.random.normal(xmax/2.0, xmax/4.0, n)
yi=np.random.normal(ymax/3.0, ymax/3.0, n)

Z, xedges, yedges = np.histogram2d(xi,yi, bins=(270,90), range=[[xmin, xmax], [ymin, ymax]])

# find the cell centers from the cell edges
x = 0.5*(xedges[:-1] + xedges[1:])
y = 0.5*(yedges[:-1] + yedges[1:])

# promote to 2D arrays
Y, X = np.meshgrid(y, x)

plt.contour(X,Y,Z)
plt.ylabel('angles')
plt.xlabel('distance')
plt.colorbar()
plt.savefig("hist2d.png")

产生这样的计数图：

但就我个人而言，在这种情况下我不会使用轮廓，因为直方图可能有噪音。

Answer 2

您的回溯表明错误不是由对 matplotlib 的调用引发的，而是 numpy 引发了 ValueError。