python - 如何过滤数据框中 2019、2020 和 2021 年的列值？

Question

我的数据框中的“REPAIR_YEAR”列包含许多不同的年份(2018 年、2019 年等)。

我想使用 pandas 定义一个新的数据框，其中根据 REPAIR_YEAR 列中的 2019 年、2020 年和 2021 年过滤所有行。

我该怎么做？

Answer 1

考虑以下数据框

[In]: df = pd.DataFrame({"REPAIR_YEAR": [2017, 2018, 2019, 2020, 2021, 2022], "Letters": ["a", "a", "a", "a", "b", "c"]})
[Out]:
   REPAIR_YEAR Letters
0   2017       a
1   2018       a
2   2019       a
3   2020       a
4   2021       b
5   2022       c

创建条件以过滤 REPAIR_YEAR 等于 2019 或 2020 或 2021 的列

condition = (df["REPAIR_YEAR"] == 2019) | (df["REPAIR_YEAR"] == 2020) | (df["REPAIR_YEAR"] == 2021)

然后按如下方式将条件应用于新数据框

[In]: new_df = df[condition]
[Out]:
   REPAIR_YEAR Letters
0   2019       a
1   2020       a
2   2021       b