Skip to main content
 首页 » 编程设计

python之读取 .csv 文件并根据特定列中的相等值连接值

2025年05月04日24jillzhang

我有一个包含三列(A、B 和 C)的 csv 文件,它们的值如下图所示: CSV Table

1,2,4 
1,257,5 
1,258,6 
1,8,7 
1,260,8 
2,24,9 
2,26,10 
2,234,11 
3,14,12 
3,22,13 
3,78,14 

我想用 "-" 连接 B 列中的值,而它们在 A 列中的值是相同的。因此,预期输出如下:

["2-257-258-8-260", "24-26-234", "14-22-78"]

任何人都可以帮助我如何获得这些结果。

提前致谢

请您参考如下方法:

这是一个普通的 Python 解决方案。

我们使用csv 阅读器来读取数据。在我的代码中,我从名为 file_data 的行列表中读取,但您可以将 file_data 更改为打开的文件对象。

我们将数据存储到列表字典中,A 列值作为键,B 列值附加到列表中。

然后我们按顺序遍历键,将 B 数据连接成所需形式的字符串。

import csv 
from collections import defaultdict 
 
file_data = '''\ 
1,2,4 
1,257,5 
1,258,6 
1,8,7 
1,260,8 
2,24,9 
2,26,10 
2,234,11 
3,14,12 
3,22,13 
3,78,14 
'''.splitlines() 
 
reader = csv.reader(file_data) 
data = defaultdict(list) 
for a, b, c in reader: 
    #print(a, b, c) 
    data[a].append(b) 
 
out = ['-'.join(data[k]) for k in sorted(data.keys())] 
print(out) 

输出

['2-257-258-8-260', '24-26-234', '14-22-78']