如何从DataFrame创建包含NumPy数组的字典？

你可以用两种简洁的方式实现需求：

方法一：字典推导式 #

直接遍历DataFrame的每一列，将列值转换为numpy数组后存入字典：

import pandas as pd
import numpy as np

# 构造示例DataFrame（和你给出的结构一致）
df = pd.DataFrame({
    'A': ['First', 'Second'],
    'B': ['row1', 'row2']
})

# 生成目标字典
d = {col: np.array(df[col]) for col in df.columns}

方法二：结合apply和to_dict #

利用apply对每一列应用np.array转换，再调用to_dict()直接生成目标结构，代码更简洁：

d = df.apply(np.array).to_dict()

这两种方法都能快速得到{列名: np.array(列值)}格式的字典，解决df.to_dict()默认输出不符合需求的问题。

内容的提问来源于stack exchange，提问作者Learning