pandas dataframe以其中一列作为键，转化为字典

import pandas as pd  

data = {  
    'key': ['A', 'B', 'C', 'D'],  
    'value1': [1, 2, 3, 4],  
    'value2': [5, 6, 7, 8]  
}  
df = pd.DataFrame(data)  
  
# 以'key'列作为键，将'value1'列作为值创建字典  
dict_value1 = df.set_index('key')['value1'].to_dict()  
print(dict_value1)  
# 输出: {'A': 1, 'B': 2, 'C': 3, 'D': 4}  
  
  
# 每个键对应多个值（例如，一个键对应一个由多个列值组成的元组或列表）：  
dict_multi_values = df.set_index('key').apply(tuple, axis=1).to_dict()  
# 或者使用列表：  
# dict_multi_values = df.set_index('key').apply(list, axis=1).to_dict()  
print(dict_multi_values)  
# 输出: {'A': (1, 5), 'B': (2, 6), 'C': (3, 7), 'D': (4, 8)}