python - 定义 CSV 的解析范围

Question

我正在导入包含 2002 年到 2011 年数据的 CSV，使用:

pair = csv.reader(open(sys.argv[1]), delimiter=' ')

names = [] ; date = [] ; open = [] ; close = [] ; min = [] ; max = []

#Parse the CSV file into a list
for data in pair:
    names.append(data[0])
names.pop(0)

我怎样才能保留 2010 年的值？ (从日期列...)

Answer 1

您测试日期，只有在正确的年份才保留它。如何做到这一点取决于日期的格式等。例如:

from datetime import datetime

for data in pair:
    date = datetime.strptime(data[1], <yourdateformat>)
    if date.year == 2010:
        names.append(data[0])