python - 我正在尝试在 python 中排序/分割 csv

Question

我得到了一个 .csv 文件，其中包含来自 11 个不同测量设备的 18k 行数据。我正在尝试为每个测量设备复制/写入一个文件，以便稍后可以绘制它们，更容易获得平均值。然而，通过这段代码，我将 YT 教程和网络资源拼凑在一起，这些文件中写入的唯一内容是“字段名称”/列的名称(无论这些内容的正确名称是什么。它只是在插入 .csv 的第一行后停止，而不是在每行中查找正确的值并将其插入到新的 .csv 文件中

我尝试使用一个 for 循环，其中有 11 个不同的 if/elif 条件，我认为这会将 device_id 列过滤到正确的设备文件。

import csv
with open('Data.csv', 'r') as Data_puntenOG:
    Data_punten = csv.DictReader(Data_puntenOG)
    for line in Data_punten:
        if line['device_id'] == 'prototype01':
            with open('HS361.csv', 'w') as HS361:        
                csv_HS361 = csv.writer(HS361)
                csv_HS361.writerow(line)
        elif line['device_id'] == "prototype02":
            with open('MinID8.csv', 'w') as MinID8:
                csv_MinID8 = csv.writer(MinID8)
                csv_MinID8.writerow(line)

然后是从prototype03到prototype12的另外9行具有不同名称/条件的相同elif行，除了9行，因为该行不在.csv文件中

11 个文件仅包含 .csv 的第一行(id、设备 ID、测量类型、测量值、时间戳)而不是一大堆带有 .csv 文件数据的行

Answer 1

如果您安装了 pandas，这将读取该文件并将具有相同“device_id”的所有行写入到一个单独的文件中，该文件的名称为“device_id”。

import pandas as pd

df = pd.read_csv('Data.csv')

EDIT:
for id in df['device_id'].unique():
    df[df['device_id'] == id].to_csv(f"{id}.csv")