新数组的总大小必须保持不变

时间:2016-09-14 10:35:36

标签: python arrays numpy reshape

我有两个数组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:新数组的总大小必须保持不变

3 个答案:

答案 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]]