这里是使用 if-else 的简单方法。
import csv
with open('input.csv') as f:
#load csv
reader = csv.reader(f)
data = [row for row in reader]
data.pop(0) #delete header
#convert
for i in range(len(data)):
print(data[i])
score_type = "MainScore" if int(data[i][1]) == 0 else "SecondScore" # <- convert it here
data[i] = [data[i][0], score_type, data[i][3], data[i][4]]
#save csv
data.insert(0, ['Source1', 'ScoreType', 'LGA', 'pred']) #add header
with open('output.csv', 'w',newline="") as f:
writer = csv.writer(f)
writer.writerows(data)
输入 csv:
Source1, Main_Score, Second_Score, LGA, pred
5, 0, 1, 1, 0.89
5, 1, 0, 1, 0.79
输出 csv:
Source1,ScoreType,LGA,pred
5,MainScore, 1, 0.89
5,SecondScore, 1, 0.79