“pandas.Series”到 numpy 数组之间的转换

Question

我有一个pandas.Series，每个元素都是一个numpy.array， 例如：

p = pandas.Series([numpy.array([1,2]), numpy.array([2,4])])

我尝试将整个 Series 转换为多维 (2,2) numpy.array，为此我使用 Series 的 values 方法，但是这个返回一个单维 numpy 数组，每个数组中的元素是numpy.array，数组的dtype是object：

In [18]: p = pandas.Series([numpy.array([1,2]), numpy.array([2,4])])

In [19]: p.values
Out[19]: array([array([1, 2]), array([2, 4])], dtype=object)

我想要达到的结果是该系列将被创建为 numpy 数组

In [23]: a = numpy.array([numpy.array([1,2]), numpy.array([2,4])])

In [24]: a
Out[24]:
array([[1, 2],
       [2, 4]])

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

有人知道如何进行这种转换吗？ to_numpy 方法对我也不起作用。

Answer 1

我建议进行以下转换：

import numpy as np
import pandas as pd

p = pd.Series([np.array([1,2]), np.array([2,4])])
np.array(p.values.tolist()).shape