将DataFrame列转换为字典键

import pandas as pd

# 创建一个示例DataFrame
data = {
    'ID': [1, 2, 3, 4],
    'Name': ['Alice', 'Bob', 'Charlie', 'David']
}
df = pd.DataFrame(data)

# 将ID列转换为字典的键，Name列作为值
dict_from_df = df.set_index('ID')['Name'].to_dict()

print(dict_from_df)

{1: 'Alice', 2: 'Bob', 3: 'Charlie', 4: 'David'}

# 示例：处理重复值
df = pd.DataFrame({
    'ID': [1, 2, 2, 3],
    'Name': ['Alice', 'Bob', 'Bobby', 'Charlie']
})

# 使用groupby聚合数据
dict_from_df = df.groupby('ID')['Name'].apply(list).to_dict()

print(dict_from_df)

{1: ['Alice'], 2: ['Bob', 'Bobby'], 3: ['Charlie']}

# 示例：确保键和值的类型一致
df['ID'] = df['ID'].astype(str)
dict_from_df = df.set_index('ID')['Name'].to_dict()

print(dict_from_df)

{'1': 'Alice', '2': 'Bob', '3': 'Charlie', '4': 'David'}