"两个包含不同顺序数据的 CSV 文件（无 Pandas）"

Question

我有两个 csv 文件来模拟我需要读取和比较的患者数据。 第一个文件是正确的患者顺序：

SUBJECT_ID,YEAR_1,YEAR_2,YEAR_3
Patient_1,131,122,142
Patient_2,109,111,112
Patient_3,111,161,112

Answer 1

我认为这就是您正在寻找的，假设您正在使用基于您发布的数据的 .csv 文件。

基本上，您可以将文件解析为 JSON，然后您可以轻松地操作它们。

import csv
import json
gender_data = []
full_data = []

with open("stack/new.csv", encoding="utf-8") as csvf:
    csvReader = csv.DictReader(csvf)
    for row in csvReader:
        gender_data.append(row)
with open("stack/info.csv", encoding="utf-8") as csvf:
    csvReader = csv.DictReader(csvf)
    for row in csvReader:
        full_data.append(row)

for x in gender_data:
    for y in full_data:
        if x["SUBJECT_ID"] == y["SUBJECT_ID"]:
            y["SEX"] = x["SEX"]

f = csv.writer(open("stack/test.csv", "w+"))

f.writerow(["SUBJECT_ID", "YEAR_1", "YEAR_2", "YEAR_3", "SEX"])

for x in full_data:
    f.writerow(
        [
            x["SUBJECT_ID"],
            x["YEAR_1"],
            x["YEAR_2"],
            x["YEAR_3"],
            x["SEX"] if "SEX" in x else "",
        ]
    )

Answer 2

您可以在不导入任何模块的情况下通过将 csv 文件读取为行列表来执行此操作，并在主文件中附加具有匹配名称的性别的行：

with open('test1.csv') as csvfile:
    main_csv = [i.rstrip() for i in csvfile.readlines()]
    
with open('test2.csv') as csvfile:
    second_csv = [i.rstrip() for i in csvfile.readlines()]
    
for n, i in enumerate(main_csv):
    if n == 0:
        main_csv[n] = main_csv[n] + ',SEX'
    else:
        patient = i.split(',')[0]
        hits = [line.split(',')[-1] for line in second_csv if line.startswith(patient)]
        if hits:
            main_csv[n] = main_csv[n] + ',' + hits[0]
        else:
            main_csv[n] = main_csv[n] + ','

with open('test.csv', 'w') as f:
    f.write('\n'.join(main_csv))