Как извлечь определенные строки из файла 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]