If statement in pandas - python

Question

I have a dataframe like this:

TechId  Rating  Category LateNight  Weekend  Weekdays 
  001     5      Teacher      1         1       0
  002     4      Student      0         1       1
  003     3      Driver       1         0       1
  004     1      Teacher      1         0       1
  005     1      Student      1         1       0
  006     0      Teacher      1         1       1
  007     2      Teacher      0         0       1

I wanted to find if the Category = Teacher is available on Weekdays, that should display TechID 003,006 & 007 in the output.

I tried this code,

Category = int(input("Enter the Trade ID you want technicians in:"))
Latenight = int(input("Press 1 if you want technicians in Latenight/else press 0:"))
Weekends = int(input("Press 1 if you want technicians in Weekends/else press 0:"))
WeekDays = int(input("Press 1 if you want technicians in WeekDays/else press 0:"))

df2 = df1_new[(df1_new['Category'] == Category) &(df1_new['Weekends'] == Weekends) & (df1_new['LateNight'] == Latenight)]
df2 = df1_new[(df1_new['Category'] == Category) &(df1_new['WeekDays'] == WeekDays) & (df1_new['LateNight'] == Latenight)]
df2 = df1_new[(df1_new['Category'] == Category) &(df1_new['WeekDays'] == WeekDays) & (df1_new['LateNight'] == Latenight) & (df1_new['Weekends'] == Weekends)]
df2_new= df2.sort_values(by=['divcount'], ascending=False)
df2_new.head()

Enter the Category you want : Teacher      
Press 1 if you want technicians in Latenight/else press 0:0
Press 1 if you want technicians in Weekends/else press 0:0
Press 1 if you want technicians in WeekDays/else press 0:1

But this displays only TechID 007 row. Some TechID might be available on both Weekend & weekdays. That should also be considered. Can anyone help me out?

I want the technicians they are available in the Latenight & Weekend or Latenight & weekend or Only Weekday. Based on the requirement. — user13424992
– user13424992, Commented May 18, 2020 at 9:43

Serge Ballesta · Accepted Answer · 2020-05-18 09:44:48Z

2

The request should ignore the filter when the answer is 0 instead of asking for 0 values. A simple way would be to use isin([0,1]) for 0 and isin([1]) for 1:

Category = int(input("Enter the Trade ID you want technicians in:"))
Latenight = [1] if int(input("Press 1 if you want technicians in Latenight/else press 0:")) else [0,1]
Weekends =[1] if  int(input("Press 1 if you want technicians in Weekends/else press 0:")) else [0,1]
WeekDays = [1] if int(input("Press 1 if you want technicians in WeekDays/else press 0:")) else [0,1]

df2 = df1_new[(df1_new['Category'] == Category) &(df1_new['WeekDays'].isin(WeekDays)) & (df1_new['LateNight'].isin(Latenight)) & (df1_new['Weekends'].isin(Weekends))]

answered May 18, 2020 at 9:44

Serge Ballesta

150k13 gold badges137 silver badges267 bronze badges

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

Comments

Abhishek Verma · Accepted Answer · 2020-05-18 09:49:10Z

0

First of all, your answer will be 004, 006 and 007.

And a short way to do this, will be:

df1_new[(df1_new['Category'] == 'Teacher') & (df1_new['Weekdays'] == 1)]

answered May 18, 2020 at 9:49

Abhishek Verma

1,7291 gold badge10 silver badges12 bronze badges

Comments

Vishal Upadhyay · Accepted Answer · 2020-05-18 10:02:49Z

0

Try this:

Teachers available on Weekdays:

df2 = df.loc[(df['Category'] == 'Teacher') & (df['Weekdays'] == 1)]
df2

Output:

TechId  Rating  Category    LateNight   Weekend Weekdays
004         1   Teacher         1           0       1
006         0   Teacher         1           1       1
007         2   Teacher         0           0       1

Teachers available on Weekdays and LateNight:

df2 = df.loc[(df['Category'] == 'Teacher') & (df['Weekdays'] & df['LateNight']==1)]
df2

Output:

TechId  Rating  Category    LateNight   Weekend Weekdays
004         1   Teacher         1           0       1
006         0   Teacher         1           1       1

edited May 18, 2020 at 10:02

answered May 18, 2020 at 9:52

Vishal Upadhyay

8191 gold badge7 silver badges19 bronze badges

Collectives™ on Stack Overflow

If statement in pandas - python

3 Answers 3

Comments

Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

3 Answers 3

Comments

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related