Pandas 将列值连接为跨列的列表，然后将整个数据帧转储到 numpy 数组中

Question

我有一个包含 10 列的数据框，每个元素作为长度 40 的列表。如何将它们转换为形状 (-1,400) 的 numpy 数组。 栏目（点击此链接查看栏目）

Answer 1

尝试： df.apply() + numpy.concatenate() + to_numpy() ：

import numpy as np

df.apply(lambda x: np.concatenate(x), axis=1).to_numpy()

测试运行

定义一个 3 行 10 列的数据框，每个元素的长度为 40：

data = [[list(range(40))] * 10] * 3
cols = list(range(10))
df = pd.DataFrame(data, columns=cols)

结果：

整体提取形状：

df.apply(lambda x: np.concatenate(x), axis=1).to_numpy().shape

(3,)

每行的形状：

df.apply(lambda x: np.concatenate(x), axis=1).to_numpy()[0].shape

(400,)