python - 将列表的一维对象 numpy 数组转换为二维数值数组并返回

Question

假设我有这个包含相同长度列表的对象数组:

>>> a = np.empty(2, dtype=object)
>>> a[0] = [1, 2, 3, 4]
>>> a[1] = [5, 6, 7, 8]
>>> a
array([[1, 2, 3, 4], [5, 6, 7, 8]], dtype=object)

1. 如何将其转换为数字二维数组？

   >>> a.shape
   (2,)
   >>> b = WHAT_GOES_HERE(a)
   >>> b
   array([[1, 2, 3, 4],
          [5, 6, 7, 8]])
   >>> b.shape
   (2, 4)
   

2. 我如何做相反的事情？

3. 如果我的a 数组是np.array 的np.array，而不是np.array of lists?

   >>> na = np.empty(2, dtype=object)
   >>> na[0] = np.array([1, 2, 3, 4])
   >>> na[1] = np.array([5, 6, 7, 8])
   >>> na
   array([array([1, 2, 3, 4]), ([5, 6, 7, 8])], dtype=object)
   

Answer 1

一种使用 np.concatenate 的方法-

b = np.concatenate(a).reshape(len(a),*np.shape(a[0]))

@Eric 建议使用 *np.shape(a[0]) 的改进应该使其适用于通用 ND 形状.

sample 运行-

In [183]: a
Out[183]: array([[1, 2, 3, 4], [5, 6, 7, 8]], dtype=object)

In [184]: a.shape
Out[184]: (2,)

In [185]: b = np.concatenate(a).reshape(len(a),*np.shape(a[0]))

In [186]: b
Out[186]: 
array([[1, 2, 3, 4],
       [5, 6, 7, 8]])

In [187]: b.shape
Out[187]: (2, 4)

---

要返回a，看来我们可以使用两步过程，就像这样-

a_back = np.empty(b.shape[0], dtype=object)
a_back[:] = b.tolist()

sample 运行-

In [190]: a_back = np.empty(b.shape[0], dtype=object)
     ...: a_back[:] = b.tolist()
     ...: 

In [191]: a_back
Out[191]: array([[1, 2, 3, 4], [5, 6, 7, 8]], dtype=object)

In [192]: a_back.shape
Out[192]: (2,)