从数据帧的列创建python字典

你要找一个 groupby + to_dict 操作-

r = df.groupby('class').name.apply(list).to_dict()

或

r = df.groupby('class').name.agg(pd.Series.tolist).to_dict()

r
{'Base': ['es', 'SS'], 'Pref': ['Ab', 'Cw'], 'Val': ['BE']}

工作原理-

• 这个 class 该列由3个类别组成。我们希望每个类别都作为索引。
• name 列需要按类分组并显示在列表中

因此 子句 操作是这里最简单的事情。分组并应用/聚合 list 得到一个具有索引的序列-

class
Base    [es, SS]
Pref    [Ab, Cw]
Val         [BE]
Name: name, dtype: object

使命感 这会给你想要的。