python - Pandas to_numpy() 生成列表数组。如何从中获取二维 numpy 数组？

Question

我有一列列表，每个列表包含相同数量的值。

如果我这样做

df['column'].to_numpy()

我得到一个列表数组:

array([list([0, 4688, 11, 43486, 40508, 13, 5,...
       list([0, 40928, 17707, 22705, 9, 38312, 2..
       list([0, 6766, 368, 3551, 28837,..
      dtype=object)

如何获取二维数组？

Answer 1

你可以这样做:

np.array(df['column'].tolist())

或者你可以简单地堆叠它们:

np.stack(df['column'].to_numpy())

这会将您的列表堆叠在一起，输出是一个二维数组。您必须确保列表的长度相同。 Numpy 数组是矩形的。