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])

Answer 2

假设日期为'％d /％m /％Y'格式

names = [data[0] for data in pair if data[1].endswith('2010')]

定义CSV的解析范围

2 个答案: