如何提取CSV文件的特定行

Question

我有一个csv excel文件，其中包含名为“ year_mo”和“ 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

我正在使用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)

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

Answer 1

这与您认为的不一样："2010" in year

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

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