Python 如何提取CSV文件的特定行

Python 如何提取CSV文件的特定行,python,csv,numpy,Python,Csv,Numpy,我有一个csv excel文件,其中包含两列,分别名为“年份”和“车辆类别” 我正在使用genfromtxt将文件加载到我的jupyter笔记本中。我的目标是提取从“2010”开始的所有行。但是,我不能这样做。以下是我的代码: filename = "demoResults.csv" data = np.genfromtxt(filename, dtype=["U50","U50"], delimiter="," ,names=True) year = data["ye

我有一个csv excel文件,其中包含两列,分别名为“年份”和“车辆类别”

我正在使用genfromtxt将文件加载到我的jupyter笔记本中。我的目标是提取从“2010”开始的所有行。但是,我不能这样做。以下是我的代码:

filename = "demoResults.csv"
data = np.genfromtxt(filename,
            dtype=["U50","U50"], delimiter="," ,names=True)

year = data["year_mo"]
year_2010 = year["2010" in year]
print(year_2010)

结果将打印出一个空列表。

这与您认为的不一样:
“2010年”

您需要做的是检查数组中的每个元素:

year_2010 = [y for y in years if "2010" in y]

您好,先生,有没有一种不使用循环的方法可以做到这一点?您需要有一个循环,问题是您需要在这个循环中执行
,列表理解是最简单的方法(IMHO)。哦,我明白了,除了使用in操作符之外,还有其他方法吗?我之所以使用它,是因为我的python知识范围仍然很小。如果要检查字符串是否在另一个字符串中,in运算符是最好的选择。还有一种方法可以在string对象上实现这一点,但它不太像Python。哦,我明白了,谢谢你的帮助!
year_2010 = [y for y in years if "2010" in y]