Python读取和编写使用日期的csv文件

Question

我想在python中看到一个代码，它读取一个包含多年历史日期的csv文件的值（每行示例：2016-09-23,2173.290039），然后这段代码会编写另一个csv文件每个日期及其在星期五发生的相关值。非常感谢你的帮助。

Answer 1

以下脚本将满足您的需求：

from datetime import datetime   
import csv

with open('input.csv', 'rb') as f_input, open('output.csv', 'wb') as f_output:
    csv_input = csv.reader(f_input)
    csv_output = csv.writer(f_output)

    for row in csv_input:
        if datetime.strptime(row[0], "%Y-%m-%d").weekday() == 4:       # if Friday
            csv_output.writerow(row)

如果input.csv包含：

2016-09-23,2173.290039
2016-09-24,2174.290039
2016-09-25,2175.290039
2016-09-26,2176.290039

它将创建包含以下内容的output.csv：

2016-09-23,2173.290039