python - 如何从另外两个 csv 文件创建新文件？

Question

我有两个 .csv 文件。

第一:
上校名称:'student_id' 和 'mark'

第二:
上校姓名:'student_id','姓名','姓氏'

我想创建第三个 .csv 文件，其中 'student_id','name', 'surname' 其中 row[ '标记'] == '五' 或 '四'

good_student=[]

for index, row in first_file.iterrows():
    if row['mark'] == 'five':
        good_student.append(row['studentId'])
    elif row['mark'] == 'four':
        good_student.append(row['studentId'])

for index, row in second_file.iterrows():
    for i in good_student:
        if row['studentId'] == i:

Answer 1

正如其他用户所建议的，数据框是处理 csv 问题的一种可靠方法。首先，我使用read_csv函数将两个csv文件读入数据帧。然后我会根据学生证加入这两个。结果是一个数据框 w第 i 列为 Student_id、mark、name 和 surname。任何缺失值都将为 NaN(调用连接的数据帧对于处理缺失值很重要)。然后，连接的数据框将根据标记单元格中的值进行过滤。

import pandas as pd

df1 = pd.read_csv('one.csv') # student_id, mark
df2 = pd.read_csv('two.csv') # student_id, name, surname

df1 = df1.join(df2.set_index('student_id'), on='student_id')

df1 = df1.loc[(df1['mark'] == 'five') | (df1['mark'] == 'four')]