Python使用Pandas库处理CSV时去除生成字典列表的列名表头方法

完整修改代码 #

import pandas
import random

word_data = pandas.read_csv("./data/german_words.csv")
# 将每条数据转换为「德语单词: 英语翻译」结构的单键字典
word_data_list = [{row['Deutsch']: row['English']} for row in word_data.to_dict(orient="records")]
print(random.choice(word_data_list))

实现说明 #

• 原有to_dict(orient="records")方法返回的每条字典会保留CSV表头作为键，不符合需求
• 新增的列表推导会遍历每条原始记录，直接用德语字段值作为键、英语字段值作为值，生成目标结构的字典列表
• 若处理的单词数据量较大，可使用itertuples遍历提高性能，写法如下：

  word_data_list = [{row.Deutsch: row.English} for row in word_data.itertuples(index=False)]
  

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