如何从其他两个 csv 文件创建新文件？

Question

我有两个 .csv 文件。

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

第二：
上校姓名：'student_id','name','surname'

我想用'student_id'、'name'、'surname' 创建第三个.csv 文件，其中row['mark'] == 'five' 或'four'

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 ith 列作为 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')]

Answer 2

您可以将两个 csv 作为数据框读取并加入它们。

import pandas as pd
df_1 = pd.read_csv("csv_1")
df_2 = pd.read_csv("csv_2")
df_1 = df_1.join(df_2)
df_1.to_csv("new_csv")

结果将是一个带有附加列的 csv 文件。如果 csv_1 的第 1 行和 csv_2 的第 1 行指的是同一事物（人、对象、ad_id...），那么它可以毫无问题地使用。

编辑： 如果两个 cvs 都按 student_id 索引行，那么最简单的方法是在加载数据帧时包含 taht：

import pandas as pd
df_1 = pd.read_csv("csv_1", index_col = "student_id")
df_2 = pd.read_csv("csv_2", index_col = "student_id")
df_1 = df_1.join(df_2)
df_1.to_csv("new_csv")