Python pandas: boolean function in Data Frame

Question

I am trying to select the rows of data frame df_A whose index values ends with 1 or 4 and capture them in another data frame df_s1s4. I am given the following hint: "pass a boolean function, which checks if an index string ends with 1 or 4, to .loc or .iloc methods."

I tried the following but couldn't get it to work.

import numpy as np

import pandas as pd

heights_A=pd.Series([176.2,158.4,167.6,156.2,161.4], index=['s1','s2','s3','s4','s5'])
weights_A=pd.Series([85.1,90.2,76.8,80.4,78.9], index=['s1','s2','s3','s4','s5'])
df_A=pd.DataFrame({'Student_height':heights_A, 'Student_weight':weights_A})
df_s1s4=df_A.loc[:,df_A.columns.str.endswith('1','4')]

print(df_s1s4)

Can anybody suggest how I might use a boolean function to solve this problem?

ernest_k · Accepted Answer · 2019-02-17 18:09:23Z

3

You can use a boolean array with .loc:

df_s1s4 = df_A.loc[(df_A.index.str.endswith('1') | df_A.index.str.endswith('4'))]

    Student_height  Student_weight
s1           176.2            85.1
s4           156.2            80.4

answered Feb 17, 2019 at 18:09

ernest_k

45.5k5 gold badges58 silver badges107 bronze badges

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

Comments

Tushar · Accepted Answer · 2021-12-15 20:15:42Z

1

Using Boolean function you can achieve results in following way:

bool_func = lambda x : x.index.str.endswith('1') + x.index.str.endswith('4')
df_s1s4 = df_A.loc[bool_func]
print(df_s1s4)

Results : 
    Student_height  Student_weight
s1           176.2            85.1
s4           156.2            80.4

answered Dec 15, 2021 at 20:15

Tushar

112 bronze badges

Comments

cottontail · Accepted Answer · 2022-07-31 01:22:29Z

0

Use this

df_s1s4=df_A.loc[df_A.index.str.endswith('1','4'),:]

edited Jul 31, 2022 at 1:22

cottontail

25.6k25 gold badges184 silver badges176 bronze badges

answered Jul 28, 2022 at 23:55

Ayush Gupta

111 bronze badge

Comments

anupamehra · Accepted Answer · 2019-07-02 21:57:59Z

-1

df_s1s4 = df_A.iloc[ [ i for i in [0, 3] if str(i)[-1] in ['0', '3'] ] ]

answered Jul 2, 2019 at 21:57

anupamehra

1

1 Comment

Rastalamm Over a year ago

Can you edit that as a code block and explain what is going on?

Collectives™ on Stack Overflow

Python pandas: boolean function in Data Frame

4 Answers 4

Comments

Comments

Comments

1 Comment

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

4 Answers 4

Comments

Comments

Comments

1 Comment

Your Answer

Sign up or log in

Post as a guest

Related