python - 从 DataFrame、Pandas、Python 创建字典

Question

我想创建一个字典，其中键是州名称，值是与该州对应的所有计数的总和。

数据帧示例

State         Count
California    100
Colorado      100
California    54
Colorado      9254

我试图使用以下内容，但不确定如何在其中包含求和函数，也不确定是否应该在此处使用 zip 方法:

df_dict = dict(zip(df.State, df.Count))
print df_dict

Answer 1

在数据框中按组计算总和，然后选择'Count'列并使用to_dict()将结果系列转换为字典:

df_dict = df.groupby('State').agg('sum')['Count'].to_dict()
print df_dict

<小时/>

如果 df 中的列数比您可能想要的两列多

df[['State','Count']].groupby('State').agg('sum')['Count'].to_dict()

避免计算所有列的总和。

编辑:

正如 EdChum 建议的那样，.agg('sum') 可以缩短为 .sum()，这会给出相同的结果。