将 numpy 数组数组转换为二维数组

Question

我有一个具有以下值的熊猫系列features （ features.values ）

array([array([0, 0, 0, ..., 0, 0, 0]), array([0, 0, 0, ..., 0, 0, 0]),
       array([0, 0, 0, ..., 0, 0, 0]), ...,
       array([0, 0, 0, ..., 0, 0, 0]), array([0, 0, 0, ..., 0, 0, 0]),
       array([0, 0, 0, ..., 0, 0, 0])], dtype=object)

现在我真的希望它被识别为矩阵，但如果我这样做

>>> features.values.shape
(10000,)

而不是(10000, 3000)这是我所期望的。

我怎样才能让它被识别为二​​维而不是一个以数组为值的一维数组。 另外为什么它不会自动将其检测为二维数组？

Answer 1

在回答您的评论问题时，让我们比较两种创建数组的方法

首先从数组列表中创建一个数组（所有长度相同）：

In [302]: arr = np.array([np.arange(3), np.arange(1,4), np.arange(10,13)])
In [303]: arr
Out[303]: 
array([[ 0,  1,  2],
       [ 1,  2,  3],
       [10, 11, 12]])

结果是一个2d数组。

相反，我们创建一个对象dtype数组，并用数组填充它：

In [304]: arr = np.empty(3,object)
In [305]: arr[:] = [np.arange(3), np.arange(1,4), np.arange(10,13)]
In [306]: arr
Out[306]: 
array([array([0, 1, 2]), array([1, 2, 3]), array([10, 11, 12])],
      dtype=object)

请注意，此显示与您的显示相同。 这是一个1d阵列。 像列表一样，它包含指向内存中其他位置的数组的指针。 请注意，它需要额外的构造步骤。 np.array的默认行为是创建一个多维数组。

需要额外的努力来解决这个问题。 同样，需要一些额外的努力才能撤消 - 创建二维数字数组。

只需在其上调用np.array就不会改变结构。

In [307]: np.array(arr)
Out[307]: 
array([array([0, 1, 2]), array([1, 2, 3]), array([10, 11, 12])],
      dtype=object)

stack确实将其更改为2d。 stack将其视为一个数组列表，它将它连接到一个新轴上。

In [308]: np.stack(arr)
Out[308]: 
array([[ 0,  1,  2],
       [ 1,  2,  3],
       [10, 11, 12]])

Answer 2

缩短@hpauli 答案：

your_2d_arry = np.stack(arr_of_arr_object)