PYTHON：从 csv 表中读取

Question

我想阅读 csv 表。到目前为止，我有一个循环，它从工作表中读取所有行并为每一行创建一个特定的 txt 文件。

with openCSV(file) as newData:
    reader = csv.reader(newData)
    next(reader)
    dictData = {}
    for i, row in enumerate(reader, 1)
        dictData = {'articleID' = row[0], 'Desc':row[1], ....}

这很好。

现在我想要的是：

如果有多个相同编号的articleID，则应将其添加到一个txt文件中。

我该怎么做？

我的 CSV 文件：

Answer 1

我认为pandas 非常适合这份工作

安装 pip3 install pandas

用法

import pandas as pd

df = pd.read_csv("data.csv")
df[df.duplicated(['ID'], keep=False)] #gives you all the duplicates

keep : {‘first’, ‘last’, False}，默认‘first’

• first ：将重复项标记为 True，但第一次出现除外。
• last ：将重复项标记为 True，但最后一次出现除外。
• False : 将所有重复项标记为 True。

Answer 2

下面。该代码创建了 3 个 csv 文件。每个 id 一个。

import csv
from collections import defaultdict

data_by_id = defaultdict(list)
with open('c:\\temp\\temp.csv')as f:
    reader = csv.reader(f)
    next(reader)
    for row in reader:
        data_by_id[row[0]].append(row[1:])

for _id, data in data_by_id.items():
    with open('c:\\temp\\{}.csv'.format(_id), 'w') as f:
        f.write('id,description,price\n')
        for line in data:
            line.insert(0, _id)
            f.write(','.join(line) + '\n')

temp.csv

id,description,price
12,desc1,12.4
13,desc13,13.4
12,desc132,312.4
13,desc1er,5.8
11,desc1wewe,77.9