python - Python：将具有 1 列的数据框转换为特定的 ndarray

Question

我有一个带有 1 列和 n 行的 pandas 数据框，例如：

                              0
0                03110311000311
1             18003130313000313
2             36003120312000312
3             54003110311000311
4             72003100310000310
...                         ...
[1400 rows x 1 columns]

我使用to_numpy()函数将此数据帧转换为 ndarray，但我得到了这个 ndarray：

我想得到一个这样的ndarray：y={ndarray:(1400,)}

怎么办？谢谢！

Answer 1

只需先选择列：

df['column_index'].to_numpy()

例子：

df = pd.DataFrame(np.arange(6))

df.to_numpy().shape            # (6,1)

df[0].to_numpy().shape       # (6,)