pandas 两列转字典

可以使用pandas中的to_dict方法将两列数据转换为字典。假设有如下的两列数据：

import pandas as pd

df = pd.DataFrame({'A': [1, 2, 3, 4, 5],
                   'B': ['a', 'b', 'c', 'd', 'e']})

可以使用下列代码将这两列数据转换为字典：

result = df.set_index('A')['B'].to_dict()
print(result)

执行上述代码后，将会得到如下结果：

{1: 'a', 2: 'b', 3: 'c', 4: 'd', 5: 'e'}

这样就将DataFrame中的两列数据A和B转换成了一个字典，其中A列的值作为字典的键，而B列的值作为字典的值。这种方法在处理数据时非常方便，可以帮助我们快速将DataFrame中的两列数据转换为字典，以便后续的操作和分析。