使用Python读写csv文件：以2022美赛C题为例

应用背景

2022美赛C题提供了比特币每日价格以及黄金价格的历史记录，分别保存在两个csv文件中。但因为黄金在某些日子不交易，因此黄金的价格在某些日期缺失，无法与比特币的每日价格对齐。使用Python则能将两个表格按日期合并起来，使得每一行显示当日的比特币价格与黄金价格。

合并前：


合并后：

csv读取

import csv
csvFile = open("LBMA-GOLDhuang.csv", "r")
reader = csv.reader(csvFile)
golddata={}
for item in reader:
    if reader.line_num==1:#此处为表头，跳过
        continue
    golddata[item[0]]=item[1]
csvFile.close()
print(golddata)

直接用open打开csv文件，之后用csv.reader(csvFile)来创建对应文件的阅读器。阅读器中的成员是一个个列表，每个列表保存了那一行的数据，以字符串形式储存。同时，为了方便把金价插入到比特币价格表中，我们使用字典存储金价，以日期为键，价格为值。

读取结果如下：

至于比特币价格，我们保留列表形式读入进来就可以了，然后以日期为键查找是否有对应的金价，如果有则加入到该行的列表中；如果没有则用0填充。

csvFile = open("BCHAIN-MKPRUbi.csv", "r")
reader = csv.reader(csvFile)
data=[]
for item in reader:
    if reader.line_num==1:
        continue
    if item[0] in golddata.keys():
        item.append(golddata[item[0]])
    else:
        item.append(0)
    data.append(item)
csvFile.close()
print(data)

输出结果可以看到一个字符串的二维列表：

csv写入

fileHeader = ["Date", "Value","USD(PM)"]#表头
csvFile=open("GoodLuck.csv","w")
writer=csv.writer(csvFile, lineterminator='\n')#设置lineterminator，否则在excel中打开csv时会多出空行
writer.writerow(fileHeader)
writer.writerows(data)
# for item in data:
#     writer.writerow(item)
csvFile.close()

同样的，使用open打开对应文件，然后用csv.writer(csvFile)声明写入器。之后就可以用writerow或writerows来写入，其中writerow的参数为一个一维列表，writerows则是二维列表。

输出完成后在Excel中打开csv文件就可以看到文章开头所呈现的效果。