Python-add tuple values to repeated keys in a di

2024-03-29 07:03:04 发布

您现在位置:Python中文网/ 问答频道 /正文

我一直在努力改变这个测试.csv文件:

a,b,4
a,c,2
b,a,4
b,c,1
b,d,5
c,a,2
c,b,1
c,d,8
c,e,10
d,b,5
d,c,8
d,e,2
d,z,6
e,c,10
e,d,2
e,z,3
z,d,6
z,e,3

转换成这种格式。到目前为止,当我尝试转换它时,重复的键被覆盖,而不是作为第二个值添加。有人能告诉我怎么得到这个格式吗?你知道吗

G1 = {
    'a': [('b', 4), ('c', 2)],
    'b': [('a', 4), ('c', 1), ('d', 5)],
    'c': [('a', 2), ('b', 1), ('d', 8), ('e', 10)],
    'd': [('b', 5), ('c', 8), ('e', 2), ('z', 6)],
    'e': [('c', 10), ('d', 2), ('z', 3)],
    'z': [('d', 6), ('e', 3)],
}

这是将文件转换为dict的代码,但会替换重复键的新值。我想知道如何将值添加为元组,对于重复的键,将元组添加为第二个值。你知道吗

reader = csv.reader(open('test.csv'))
arcos = {}
for row in reader:
    key = row[0]
    arcos[key] = row[1:]

结果是:

{
'a': ['c', 2.0],
 'b': ['d', 5.0],
 'c': ['e', 10.0],
 'd': ['z', 6.0],
 'e': ['z', 3.0],
 'z': ['e', 3.0]
}

我希望我所要求的是可能的。你知道吗


Tags: 文件csvkey代码test格式opendict
2条回答

您可以使用groupby方法来简化:

from itertools import groupby

with open('out.csv') as f:
    print({k:[(i.split(',')[1],i.split(',')[2]) for i in g] for k, g in groupby([i.strip() for i in f],lambda x: x.split(',')[0])})

结果:

 {'a': [('b', '4'), ('c', '2')], 'c': [('a', '2'), ('b', '1'), ('d', '8'), ('e', '10')], 'b': [('a', '4'), ('c', '1'), ('d', '5')], 'e': [('c', '10'), ('d', '2'), ('z', '3')], 'd': [('b', '5'), ('c', '8'), ('e', '2'), ('z', '6')], 'z': [('d', '6'), ('e', '3')]}

使用defaultdictlist类,迭代文件并附加元组:

from collections import defaultdict

G = defaultdict(list)

with open('test.csv') as file:
    for line in file:
        key, letter, digit = line.strip().split(',')
        digit = int(digit)
        G[key].append((letter, digit))

相关问题 更多 >