numpy 数组 1.9.2 获取 ValueError：无法将输入数组从形状（4,2）广播到形状（4）

Question

以下代码在 numpy 1.7.1 中运行，但在当前版本中给出值错误。我想知道它的根本原因。

    import numpy as np
    x = [1,2,3,4]
    y = [[1, 2],[2, 3], [1, 2],[2, 3]]

    a = np.array([x, np.array(y)])

以下是我在 numpy 1.7.1 中得到的输出

>>>a
array([[1, 2, 3, 4],
       [array([1, 2]), array([2, 3]), array([1, 2]), array([2, 3])]], dtype=object)

但相同的代码在 1.9.2 版本中会产生错误。

    ----> 5 a = np.array([x, np.array(y)])

ValueError: could not broadcast input array from shape (4,2) into shape (4) 

我找到了一种可能的解决方案。但我不知道这是否是最好的做法。

b= np.empty(2, dtype=object)
b[:] = [x, np.array(y)]

>>> b
array([[1, 2, 3, 4],
       array([[1, 2],
       [2, 3],
       [1, 2],
       [2, 3]])], dtype=object)

请提出一个解决方案来实现所需的输出。谢谢

Answer 1

您到底想生产什么？我没有 1.7 版本来测试您的示例。

np.array(x) 产生一个(4,) 数组。 np.array(y)(4,2)。

如评论中所述，在 1.8.1 np.array([x, np.array(y)]) 中产生

ValueError: setting an array element with a sequence.

我可以制作一个object dtype数组，由列表和数组组成

In [90]: np.array([x, np.array(y)],dtype=object)
Out[90]: 
array([[1, 2, 3, 4],
       [array([1, 2]), array([2, 3]), array([1, 2]), array([2, 3])]], dtype=object)

我还可以将 2 个数组连接成一个 (4,3) 数组（x 是第一列）

In [92]: np.concatenate([np.array(x)[:,None],np.array(y)],axis=1)
Out[92]: 
array([[1, 1, 2],
       [2, 2, 3],
       [3, 1, 2],
       [4, 2, 3]])

np.column_stack([x,y]) 做同样的事情。

---

奇怪的是，在 dev 1.9（我没有安装生产 1.9.2）中它可以工作（有点）

In [9]: np.__version__
Out[9]: '1.9.0.dev-Unknown'

In [10]: np.array([x,np.array(y)])
Out[10]: 
array([[        1,         2,         3,         4],
       [174420780, 175084380,  16777603,         0]])
In [11]: np.array([x,np.array(y)],dtype=object)
Out[11]: 
array([[1, 2, 3, 4],
   [None, None, None, None]], dtype=object)
In [16]: np.array([x,y],dtype=object)
Out[16]: 
array([[1, 2, 3, 4],
   [[1, 2], [2, 3], [1, 2], [2, 3]]], dtype=object)

所以看起来有某种发展正在进行中。

在任何情况下，从这个列表中创建一个新数组和一个二维数组都是模棱两可的。使用column_stack（假设你想要一个二维整数数组）。

---

numpy 1.9.0 发行说明：

改进了使用 np.array 将包含数组的列表转换为数组的性能。现在它的速度与 np.vstack(list) 相当。

转置y vstack 有效：

In [125]: np.vstack([[1,2,3,4],np.array([[1,2],[2,3],[1,2],[2,3]]).T])
Out[125]: 
array([[1, 2, 3, 4],
       [1, 2, 1, 2],
       [2, 3, 2, 3]])

如果 1.7.1 有效，并且 x 是字符串名称，而不仅仅是您的示例中的整数，那么它可能正在生成一个对象数组。