Pandas, string case statement to dataframe - More then two conditions

Asked Jun 10 '19 at 15:02

Active Jun 10 '19 at 19:49

Viewed 18 times

I have a dataframe where I am trying to insert a column and have what is equivalent to a case statement based on another column within the dataframe.

Case statement logic:

Case When column a IN ('Stack','Overflow') THEN 'St' WHEN column a IN ('Super','User') THEN 'Su'

what I have:

final_df["business_name"] = np.where[(final_df['column a'].isin(('Stack', \
                                                'Overflow')),'St', "")

I want to add an & to do another condition.

& (final_df['column a'].isin(('Super')), 'Su', "")]

Sample data:

+----------+----------------+
| column a | business_name  |
+----------+----------------+
| Stack    | St             |
| Overflow | St             |
| Super    | Su             |
| User     | Su             |
+----------+----------------+

edited Jun 10 '19 at 19:49

asked Jun 10 '19 at 15:02

excelguy

1,574
6
33
67

It feels like you've asked this before. – cs95 Jun 10 '19 at 15:03
yes i have, thought i would give it another shot. – excelguy Jun 10 '19 at 15:05
You posted a comment under an answer of mine. That definitely should have worked for you. – cs95 Jun 10 '19 at 15:05
i asked the wrong question – excelguy Jun 10 '19 at 15:21
use `np.where(df['column a'].isin(('Stack','Overflow')),'St','Su')` or `np.select([df['column a'].isin(('Stack','Overflow')),df['column a'].isin(('Super','User'))],['St','Su'])` – anky Jun 10 '19 at 16:35
thanks for this. What if I have more then 2 conditions? – excelguy Jun 10 '19 at 19:08

Pandas, string case statement to dataframe - More then two conditions

0 Answers0