从文件中创建一个包含列表的嵌套字典

Question

例如对于txt文件的

Math, Calculus, 5 
Math, Vector, 3
Language, English, 4
Language, Spanish, 4

进入字典：

 data={'Math':{'name':[Calculus, Vector], 'score':[5,3]}, 'Language':{'name':[English, Spanish], 'score':[4,4]}}

我在附加值以在较小的字典中创建列表时遇到问题。我对此很陌生，我不会理解导入命令。非常感谢您的所有帮助！

Answer 1

根据您的数据：

data={'Math':{'name':['Calculus', 'Vector'], 'score':[5,3]}, 
      'Language':{'name':['English', 'Spanish'], 'score':[4,4]}}

如果你想追加到字典中的列表，你可以这样做：

data['Math']['name'].append('Algebra')

data['Math']['score'].append(4)

如果你想添加一个新的字典，你可以这样做：

data['Science'] = {'name':['Chemisty', 'Biology'], 'score':[2,3]}

我不确定这是否是您想要的，但我希望它有所帮助！

Answer 2

对于每一行，找到 3 个值，然后将它们添加到 dict 结构中

from pathlib import Path

result = {}
for row in Path("test.txt").read_text().splitlines():
    subject_type, subject, score = row.split(", ")

    if subject_type not in result:
        result[subject_type] = {'name': [], 'score': []}

    result[subject_type]['name'].append(subject)
    result[subject_type]['score'].append(int(score))

---

如果密钥不存在，您可以使用 defaultdict 创建映射来简化它

result = defaultdict(lambda: {'name': [], 'score': []}) # from collections import defaultdict
for row in Path("test.txt").read_text().splitlines():
    subject_type, subject, score = row.split(", ")
    result[subject_type]['name'].append(subject)
    result[subject_type]['score'].append(int(score))

---

使用pandas.DataFrame可以直接格式化数据，输出你想要的格式

import pandas as pd

df = pd.read_csv("test.txt", sep=", ", engine="python", names=['key', 'name', 'score'])
df = df.groupby('key').agg(list)
result = df.to_dict(orient='index')