Pandas 数据框到二维 numpy 数组

Question

我有以下数据框:

d = {'histogram' : [[1,2],[3,4],[5,6]]}
df = pd.DataFrame(d)

直方图的长度始终相同(在本示例中为 2)。

我想将“直方图”列转换为 2D numpy 数组以馈入神经网络。首选输出是:

output_array = np.array(d["histogram"])

即:

array([[1, 2],
       [3, 4],
       [5, 6]])

然而当我尝试时:

df["histogram"].to_numpy()

结果是一个列表数组而不是数组的 numpy 数组:

array([list([1, 2]), list([3, 4]), list([5, 6])], dtype=object)

这对神经网络来说是有问题的，因为我必须指定尺寸/形状。

我尝试通过转换为 numpy 数组来解决问题:

df["histogram_arrays"] = df["histogram"].apply(lambda x: np.array(x))
df["histogram_arrays"].to_numpy()

返回一维数组而不是二维数组。

array([array([1, 2]), array([3, 4]), array([5, 6])], dtype=object)

如何将直方图放入二维数组中？

Answer 1

试试这个:

np.vstack(df['histogram'])