我有两个数组x1和x2,两个都是1 * 14个数组我试图将它们压缩然后执行重塑。
代码如下;
x1
输出[122]:数组([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(最近一次调用最后一次) in() ----> 1 X = np.array(zip(x1,x2))。reshape(2,len(x1))
ValueError:新数组的总大小必须保持不变
答案 0 :(得分:3)
我会假设您使用的是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那样返回迭代器,并且您以前的代码可以正常工作。
答案 1 :(得分:3)
您使用的是Python 3,因此zip
会被懒惰地评估。
>>> np.array(zip(x1,x2))
array(<zip object at 0x7f76d0befe88>, dtype=object)
你需要迭代它:
>>> np.array(list(zip(x1, x2))).reshape(2, len(x1))
array([[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]])
答案 2 :(得分:1)
np.array
无法将zip
创建的生成器识别为可迭代的。如果您先强制转换为列表,它可以正常工作:
from array import array
import numpy as np
x1 = array('i', [1, 2, 3, 1, 5, 6, 5, 5, 6, 7, 8, 9, 7, 9])
x2 = array('i', [1, 3, 2, 2, 8, 6, 7, 6, 7, 1, 2, 1, 1, 3])
print(np.array(list(zip(x1, x2))).reshape(2, len(x1)))
给出
[[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]]