Создание словаря из DataFrame, Pandas, Python

Question

Я хочу создать словарь, в котором ключ является именем состояния, а значение представляет собой сумму всего количества, соответствующего состоянию.

Пример из dataFrame

State   Count 
California 100 
Colorado  100 
California 54 
Colorado  9254 

Я пытался использовать следующее, но не знаете, как включить функцию суммы внутри этого, и не уверен, если я должен использовать метод зип здесь:

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

Answer 1

Вычислить сумму по группе в dataframe, а затем выберите 'Count' колонки и использовать to_dict() преобразовать получившиеся серии словаря:

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

---

Если у вас есть несколько столбцов в ФРЕ, чем те два вы, вероятно, захотите сделать

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

, чтобы избежать вычисления суммы для всех столбцов.

Edit:

, как EdChum предложил .agg('sum') может быть сокращен до .sum(), который дает тот же результат.

Answer 2

Это не сработает должным образом, так как конструктор dict заменит значение каждого state, как оно идет, вместо того, чтобы суммировать. Это не один лайнер, но:

from collections import Counter 

df_dict = Counter() 
for state, count in zip(df.State, df.Count): 
    df_dict[state] += count 

получит Вас общий счет, а Counter класса имеет некоторые дополнительные функциональные возможности, которые могут быть полезны для словарей графов.