2d直方图:值错误:要解压缩的值太多(预期为 2)

Question

我正在尝试从散点图创建二维直方图。但是我使用以下代码获得了error: ValueError: too many values to unpack (expected 2)

如果我更改输入数据以包含 xy 坐标的一个列表，它可以正常工作。如果我只选择 2dhistogram 行中的第一个列表，它也有效。例如 zi, xi, yi = np.histogram2d(x[0], y[0], bins=bins) .

import matplotlib.pyplot as plt
import numpy as np
import random
from functools import partial
##create list of lists (x,y coordinates)
x_list = partial(random.sample, range(80), 10)
y_list = partial(random.sample, range(80), 10)
x = [x_list() for _ in range(10)]
y = [y_list() for _ in range(10)]
fig, ax = plt.subplots()
ax.set_xlim(0,80)
ax.set_ylim(0,80)
bins = [np.linspace(*ax.get_xlim(), 80),
        np.linspace(*ax.get_ylim(), 80)]
##error occurs in this line
zi, xi, yi = np.histogram2d(x, y, bins=bins)
zi = np.ma.masked_equal(zi, 0)
ax.pcolormesh(xi, yi, zi.T)    
ax.set_xticks(bins[0], minor=True)
ax.set_yticks(bins[1], minor=True)
ax.grid(True, which='minor')
scat = ax.scatter(x, y, s = 1)

我能找到的唯一一篇关于这个的帖子建议尝试将 x，y 更改为 numpy 数组。我试过这个，但仍然得到相同的错误代码。

zi, xi, yi = np.histogram2d(np.asarry(x), np.asarray(y), bins=bins)

还有其他建议吗？

Answer 1

np.histogram2d需要x和y坐标的平面列表，而不是列表列表。您可以非常轻松地解决此问题。只需将填充x和y的行更改为扁平化列表推导

：

x = [num for _ in range(10) for num in x_list()]
y = [num for _ in range(10) for num in y_list()]

或者，您可以跳过使用 random.sample 和 partial 的全部复杂性，而只使用 np.random.randint，这可以创建任何给定形状的随机整数数组：

x = np.random.randint(0, 80, size=100)
y = np.random.randint(0, 80, size=100)