Pandas/Python check value in between and output value form another column

Question

I have the following dataframe, and a variable equal to 20. I need to find where this variable is in col2 (here between 10 and 50) and output the lowest value in col1, here would be 2. Another example would be to have a variable equal to 51 and the correct output of col1 would be 3.

I could do it with an if formula but I am hopping for a better solution for larger datasets.

d = {'col1': [1, 2, 3, 4, 5], 'col2': [0, 10, 50, 60, 70]}
df = pd.DataFrame(data=d)

jezrael · Accepted Answer · 2021-03-19 09:03:23Z

1

Use boolean indexing with DataFrame.loc and then get last value of column col1 by Series.iat:

a = df.loc[df['col2'] <= 20, 'col1'].iat[-1]
print (a)
2
a = df.loc[df['col2'] <= 51, 'col1'].iat[-1]
print (a)
3
a = df.loc[df['col2'] <= 10, 'col1'].iat[-1]
print (a)
2

answered Mar 19, 2021 at 9:03

jezrael

868k103 gold badges1.4k silver badges1.3k bronze badges

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

Collectives™ on Stack Overflow

Pandas/Python check value in between and output value form another column

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