如何用Python将DataFrame标题列单词设为字典键、类别为值？

完全可以实现，以下是具体的代码方案：

步骤1：构造示例DataFrame（已有数据集可跳过） #

先导入pandas并创建你提供的示例数据集：

import pandas as pd

data = {
    'Title': ["Harry Potter and the Sorcerer's Stone", "The Matrix Movie series", "Diary of a Wimpy Kid"],
    'Category': ["Fiction", "Sci-Fi", "Comics"]
}
df = pd.DataFrame(data)

步骤2：生成单词-分类映射字典 #

遍历DataFrame的每一行，拆分Title列的单词并与对应Category绑定：

word_category_map = {}

# 遍历每一行数据
for _, row in df.iterrows():
    # 按空格拆分标题为单个单词
    title_words = row['Title'].split()
    current_category = row['Category']
    # 将每个单词作为键，分类作为值存入字典
    for word in title_words:
        word_category_map[word] = current_category

步骤3：查看结果 #

如果需要按示例格式打印结果，执行以下代码：

for word, category in word_category_map.items():
    print(f"{word}: {category}")

注意事项 #

• 若不同行的Title中出现重复单词，字典中该单词对应的分类会被最后一次遍历到的行的Category覆盖。
• split() 默认按空格拆分单词，像Sorcerer's这类带特殊符号的完整单词会被保留为一个键。

内容的提问来源于stack exchange，提问作者user19287021