python - 基于列条件的 Pandas Dataframe 过滤

Question

我正在尝试从数据框中删除其中一列满足特定条件的所有行。

假设我的数据框是这样的:

index  'Value 1'   'Grade'
1         10         2170A
2         15         2170B
3         10         NCR2170B
4         20         NCR2170A
5         30         NCR2170B

我希望过滤后的输出是这样的

    index  'Value 1'   'Grade'
2         15         2170B
3         10         NCR2170B
5         30         NCR2170B

我尝试了使用列表理解的许多不同变体，循环遍历每一行并进行评估。当我使用以下内容时，我可以让它工作:

data_filtered = data[data['Grade'] == '2170B']

但这显然错过了像 NCR2170B 这样的条目。

每当我尝试类似的事情时:

data_filtered = data['2170B' in data['Grade']]

我收到关键错误:“True”。

我觉得我在这里遗漏了一些非常明显的东西。

我也尝试过使用 np.where 但它只输出一个空数组。

Answer 1

您可以使用str.contains检查字符串列是否包含子字符串:

df[df.Grade.str.contains('2170B')]

#index  Value 1    Grade
#1  2        15    2170B
#2  3        10 NCR2170B
#4  5        30 NCR2170B