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