python - 2 个 numpy 二维数组的散点图

Question

我有 2 个 numpy 二维数组

import numpy as np
arr1 = np.random.rand(4,4)
arr2 = np.random.rand(4, 4)

我想显示一个散点图，其中 x 轴值来自 arr1，y 轴值来自 arr2。我试过这个:

s = (arr1.ravel(), arr2.ravel())
x, y = zip(*s)
plt.scatter(x, y)

但我收到此错误:

ValueError: too many values to unpack (expected 2)

如何解决这个问题？

Answer 1

IIUC，您不需要 zip 步骤:

s = (arr1.ravel(), arr2.ravel())
plt.scatter(*s)
plt.show()

或者，只需绘制 arr1 和 arr2 即可得到相同的结果:

plt.scatter(arr1, arr2)
plt.show()

原因是通过压缩，您将创建大量坐标元组:

>>> list(zip(*s))
[(0.5233576831070681, 0.3622905772322086), (0.6771459051981418, 0.46550530512072197), (0.5836238435117137, 0.9678582465614765), (0.5215640961420361, 0.05930220671158393), (0.0671420867491811, 0.3229294577997751), (0.09453539420217905, 0.5853394743453684), (0.14945382971858256, 0.7100764985571005), (0.6097216935812977, 0.008285702808878748), (0.8008615396938863, 0.8564656781773946), (0.0012100717276920525, 0.03915037044723313), (0.5749921053520953, 0.43147487440006294), (0.5965950855156836, 0.6485170240649151), (0.00223226469849902, 0.3134990067863225), (0.6424325871844799, 0.4041957463865189), (0.06797409254523168, 0.685192515451394), (0.9485129458199039, 0.6873427463294267)]

这是正确的(尽管它们不能打包成 2 个变量，因为它是 16 个坐标对)，但是您基本上会想要迭代来绘制它们，这并不理想。或者您可以重新压缩它们，但我认为这不是一个有效的方法:

x,y = zip(*zip(*s))
plt.scatter(x,y)
plt.show()