python - 新数组的总大小必须不变

Question

我有两个数组 x1 和 x2，都是 1*14 数组，我试图将它们压缩然后执行 reshape 。

代码如下;

x1

Out[122]: array([1, 2, 3, 1, 5, 6, 5, 5, 6, 7, 8, 9, 7, 9])

x2

输出[123]: 数组([1, 3, 2, 2, 8, 6, 7, 6, 7, 1, 2, 1, 1, 3])

X = np.array(zip(x1, x2)).reshape(2, len(x1))

ValueErrorTraceback(最后一次调用) 在 ()----> 1 X = np.array(zip(x1, x2)).reshape(2, len(x1))

ValueError:新数组的总大小必须不变

Answer 1

我假设您使用的是 Python 3，其中结果是一个包含 zip 对象的数组。

您应该对压缩 项目调用list:

X = np.array(list(zip(x1, x2))).reshape(2, len(x1))
#            ^^^^
print(X)
# [[1 1 2 3 3 2 1 2 5 8 6 6 5 7]
#  [5 6 6 7 7 1 8 2 9 1 7 1 9 3]]

在 Python 2 中，zip 返回一个列表而不是 Python 3 中的迭代器，您之前的代码可以正常工作。