我有两个巨大的CSV文件,并希望他们使用python pandas加入一个新的CSV文件,主键是id_student,我可以成功加入不同的列,但是当我输出到一个新的CSV文件时,整个束数据只存在于第一行,不同的列,例如,第1行第1列将是id_student,它就像:
0 12345
1 12344
然后第1行将是final_result,格式如下:
0 Pass
1 Pass
但我的预期输出将是:
0 12345 Pass
1 12344 Pass
有什么方法可以修复输出格式吗?
def plotlyGraph(self):
df = pandas.read_csv('studentAssessment.csv')
dc = pandas.read_csv('studentInfo.csv')
res = pandas.merge(df,dc, on=['id_student'], how='outer')
a=res['id_student']
b=res['final_result']
c=res['score']
d=res['id_assessment']
e=res['region']
with open("new.csv", "w", newline="") as csvfile:
writer = csv.writer(csvfile)
writer.writerow([a,b,c,d,e])
我假设你的df
有2列:id_student
和id_assessment
,而dc
有2列:id_student
和final_result
。试试这个:
df = pandas.read_csv('studentAssessment.csv')
dc = pandas.read_csv('studentInfo.csv')
res = df.merge(dc, on=['id_student'], how='outer')
print(res)
产量
id_student id_assessment final_result
0 0 12345 pass
1 1 12344 pass
要存储在csv
文件中:
res.to_csv("new.csv", index=False)