如何使用Python在大型csv文件上打印特定年份

Question

import pandas

fileref = open('chart.csv')
f = pandas.read_csv(fileref)

f_set = f[f.year >= 2005]
print(f_set.groupby('y').namefromchart.nunique())

所以我知道f_set将打印出我的y单元格与2005年的相应年份。 如果我想打印出2002年至2009年的内容怎么办？

Answer 1

看来你需要为boolean indexing添加第二个条件：

import pandas as pd

f = pd.read_csv('chart.csv')
f_set = f[(f.year >= 2002) & (f.year < 2010)]
print (f_set.groupby('y').namefromchart.nunique())

默认情况下， between另一个解决方案包括 ：

f_set = f[f.year.between(2002,2009)]

Answer 2

考虑pd.DataFrame f

f = pd.DataFrame(dict(year=range(2000, 2011), A=np.random.rand(11)))

你可以设置索引并切片你喜欢的方式

f.set_index('year').ix[2002:2009]