来自具有两个键和多个值的文件的 Python 字典

Question

（这个问题已经在 stackoverflow 上的几篇文章中得到解答。但是，我无法得到正确的结果，我不知道我做错了什么？）

我想从一个包含两个键和 14 个值的文本文件中创建一个字典：

data.txt:

Key1    Key2    Val1    Val2    Val3…Val14
100       a     x0      y0      z0………n0
101       a     x1      y1      z1………n1
102       b     x2      y2      z2………n2
103       b     x3      y3      z3………n3
104       c     x4      y4      z4………n4
105       c     x5      y5      z5………n5
…
140       m     xm      ym      zm………nm

字典应该是这样的：

{100: {a: [x0, y0, z0,…n0]},
101: {a: [x1, y1, z1,…n1]},
102: {b: [x2, y2, z2,…n2]},
103: {b: [x3, y3, z3,…n3]},
 …
140: {m: [xm, ym, zm,…nm]}}

我已经尝试过 Code1 和 Code2。 Code1 提供了一个非常大的字典，其中的行与附加的其他行重复。 Code2 给出错误 TypeError: unhashable type: 'slice'。

Code1:
lookupfile = open("data.txt", 'r')
lines = lookupfile.readlines()
lookup = lines[1:]   # Start the dictionary from row 1, exclude the column names
d={}
for line in lookup:
    dic = line.split()
    d.update({dic[0]: {dic[1]: dic[2:]}})
    print(d) 

Code2:
data = defaultdict(dict)
with open('data.txt', 'r') as file:
    reader = csv.DictReader(file)
    for row in reader:
        data[row['Key1']][row['Key2']]=row['Val1':]
        print (data)

我希望代码看起来像 Code2，这样我以后可以使用列名。 但是，我将不胜感激。

如果需要，我可以提供更多信息。

Answer 1

s="""Key1    Key2    Val1    Val2    Val3…Val14
100       a     x0      y0      z0
101       a     x1      y1      z1
102       b     x2      y2      z2
103       b     x3      y3      z3
104       c     x4      y4      z4
105       c     x5      y5      z5"""
d  = {}
for line in s.splitlines()[1:]:
    spl = line.split()
    d[spl[0]] ={spl[1]:spl[2:]}

from pprint import pprint
pprint(d)
{'100': {'a': ['x0', 'y0', 'z0']},
 '101': {'a': ['x1', 'y1', 'z1']},
 '102': {'b': ['x2', 'y2', 'z2']},
 '103': {'b': ['x3', 'y3', 'z3']},
 '104': {'c': ['x4', 'y4', 'z4']},
 '105': {'c': ['x5', 'y5', 'z5']}}

相同的逻辑适用于您的文件代码，以跳过文件对象上的第一行调用next。然后像上面那样简单地索引每一行。

d = {}
with open('data.txt', 'r') as f:
    next(f) # skip header
    for row in f:
        spl = line.split()
        # slicing using spl[2:] will give you a list of all remaining values
        d[spl[0]] = {spl[1]:spl[2:]}

如果您的列之间实际上有多个空格，使用 str.split 会比使用 csv 模块更好。

Answer 2

您正在使用DictReader，因此每个row 都是dict，并且您不能对dict 进行切片（就像您在作业的RHS 上尝试做的那样）。

所以使用一个普通的csv.reader（所以每个row都是一个list，你可以对其进行切片）并且：

data[row[0]][row[1]]=row[2:]