Python Pandas Get Values According to If/Else

Question

My input dataframe;

Order    Need   WarehouseStock   StoreStock
1        3      74               5
0        4      44               44
0        0      44               44
6        12     44               44
0        6      644              44
6        6      44               44

I want to count whether any difference or not among "Order" and Need values with below code;

difference = df['Need'] - df['Order']
mask = difference.between(-1,1)                                                              
print (f'Count: {(~mask).sum()}')

I want to that something like this;

If (WarehouseStock-StoreStock) >= Need:

difference1 = df['Need'] - df['Order']
mask1 = difference1.between(-1,1)                                                              
print (f'Count: {(~mask1).sum()}')

Else

difference2 = df['Need'] - df['Order']
mask2 = difference2.between(-5,5)                                                              
print (f'Count: {(~mask2).sum()}')

Desired Outputs are;

Count 3

Order    Need   WarehouseStock   StoreStock
1        3      74               5
6        12     44               44
0        6      644              44

Could you please help me about this?

Chris · Accepted Answer · 2019-10-23 12:01:12Z

4

Using numpy.where with pandas.Series.between:

import pandas as pd
import numpy as np

s = df['Need'] - df['Order']
ind = np.where((df['WarehouseStock'] - df['StoreStock']).ge(df['Need']), ~s.between(-1, 1), ~s.between(-5 , 5))

Output:

ind.sum()
# 3

df[ind]
   Order  Need  WarehouseStock  StoreStock
0      1     3              74           5
3      6    12              44          44
4      0     6             644          44

answered Oct 23, 2019 at 12:01

Chris

29.8k3 gold badges34 silver badges56 bronze badges

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

Collectives™ on Stack Overflow

Python Pandas Get Values According to If/Else

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