Python dataframe 某两列变成一个dict

1、把dataframe中的某两列变成一个key-value形式的dict

      代码如下：

import pandas as pd

df = pd.DataFrame({'A': ['a', 'b', 'c'],
                   'B': [2, 7, 1],
                   'C': [100, 87, 96]})

# 形成以A为key,B为value的dict
dc = df.set_index("A")["B"].to_dict()
print(dc)

结果如下：

{'a': 2, 'b': 7, 'c': 1}

2、一个key对应多个value的情况

dc = df.groupby('word')['count'].apply(list).to_dict()  # 对于同一个key对应多个value，则把同一key的value构成一个list