Selecting specific columns with conditions using python pandas

Question

In my Dataframe, I would like to choose only specific columns based on a certain condition from a particular column.

Here is my output of initial Data Frame using df.head()

I would like to find for column equals to 'B' and display it with selected columns.

My code:

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']]

How to merge the condition and display it with selected columns only.

Thanks

df[df['diagnosis'] == 'B'][['diagnosis','radius_mean','perimeter_mean','area_mean']]. — Quang Hoang
– Quang Hoang, Commented Apr 9, 2019 at 17:08

Mohit Motwani · Accepted Answer · 2019-04-09 17:08:39Z

3

Use df.loc:

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

answered Apr 9, 2019 at 17:08

Mohit Motwani

4,8124 gold badges21 silver badges50 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

Selecting specific columns with conditions using python pandas

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related