如何提取CSV文件的特定行

Question

我有一个csv excel文件，其中包含两列，分别命名为"year_mo"one_answers"vehicle_class"。

year_mo, vehicle_class
2010-01, Category A
2010-02, Category B
2010-03, Category C
2011-01, Category A
2011-02, Category B
2011-03, Category C
2011-04, Category D
2011-05, Category E

我正在使用genfromttxt将文件加载到我的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)

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

Answer 1

这不会做你认为它会做的事情："2010" in year

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

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