Perform apply function in multiple column using pandas

Question

I am trying to find rows where none of the three actors' Facebook likes should be less than half of the other two.

But i am getting an error KeyError: 'actor_1_facebook_likes'.Could you please help me with this.

I think you need popular_trio.apply(get_correct_trio, axis=1) — jezrael
– jezrael, Commented Dec 21, 2020 at 7:46
Please take the time to read this post on how to provide a great pandas example as well as how to provide a minimal, complete, and verifiable example and revise your question accordingly. These tips on how to ask a good question may also be useful. — jezrael
– jezrael, Commented Dec 21, 2020 at 7:49

jezrael · Accepted Answer · 2020-12-21 08:18:43Z

1

I wanted to show you a much cleaner and performant way that you can write the code. Create the conditions and use ~ to drop the rows that meet those conditions. I hope this helps:

a1 = popular_trio['actor_1_facebook_likes']
a2 = popular_trio['actor_2_facebook_likes']
a3 = popular_trio['actor_3_facebook_likes']

c1 = (a1/2) < a2
c2 = (a2/2) < a1
c3 = (a3/2) < a2
c4 = (a1/2) < a3
c5 = (a2/2) < a3
c6 = (a3/2) < a1

popular_trio[~(c1|c2|c3|c4|c5|c6)] # or try popular_trio[~((c1)|(c2)|(c3)|(c4)|(c5)|(c6))]

edited Dec 21, 2020 at 8:18

jezrael

867k102 gold badges1.4k silver badges1.3k bronze badges

answered Dec 21, 2020 at 8:06

David Erickson

16.7k2 gold badges21 silver badges37 bronze badges

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

1 Comment

David Erickson Over a year ago

@jezrael yes, I know. I just couldn't help myself to clean up the code ;)

Collectives™ on Stack Overflow

Perform apply function in multiple column using pandas

1 Answer 1

1 Comment

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

1 Comment

Related