python-3.x - 使用 python pandas 选择具有条件的特定列

Question

在我的数据框中，我只想根据特定列中的特定条件选择特定列。

这是我使用 df.head() 输出的初始数据框

我想找到等于“B”的列并将其与选定的列一起显示。

我的代码：

df = pd.read_csv('cancer_data.csv')

#To display column diagnosis equals B
df[df['diagnosis'] == 'B']

#To display selected columns
df[['diagnosis','radius_mean','perimeter_mean','area_mean']]

如何合并条件并将其仅与选定的列一起显示。

谢谢

标签： python-3.xpandas

Answer 1

Use df.loc:

df.loc[df['diagnosis'] == 'B', ['diagnosis','radius_mean','perimeter_mean','area_mean']]