Implement own function

Question

I am trying to implement my own function with the data set below:

import pandas as pd
import numpy as np
    
data = { 
            'sales': ['0','1','2','2','6','5','6'],            
           }        
df = pd.DataFrame(data, columns = ['sales'])
    
df

Now I want to apply my function how will give 1 only on value '2', 2 only on values '6', and for all others will give '3'. In order to do this, I try this function :

def function_test(data):
    sales = df['sales']
    
    conditions = [
                 (sales == '6'), 
                 (sales == '2'),
                 (sales <> '6'&'2') #<----This row
                 
                 ]
    values = [ 
             1,
             2,
             3
             ]
    dummy = np.select(conditions, values)    
    return (dummy)

But this function has a problem for third conditions, so can anybody help me how to solve this problem?

sales not in (1,2) ?

Patrick Artner
– Patrick Artner

2022-04-18 19:25:38 +00:00
Commented Apr 18, 2022 at 19:25 — Patrick Artner
– Patrick Artner, Commented Apr 18, 2022 at 19:25

constantstranger · Accepted Answer · 2022-04-18 21:34:04Z

1

One way to fix it is to use != instead of <> and use two comparisons. (I also changed the condition sequence to match what you described in the text of your question):

((sales != '6') & (sales != '2'))

Full test code:

def function_test(data):
    sales = df['sales']
    conditions = [
                 (sales == '2'),
                 (sales == '6'), 
                 ((sales != '6') & (sales != '2'))
                 ]
    values = [1, 2, 3]
    dummy = np.select(conditions, values)    
    return (dummy)

print(function_test(data))

Results:

[3 3 1 1 2 3 2]

edited Apr 18, 2022 at 21:34

answered Apr 18, 2022 at 19:38

constantstranger

9,4072 gold badges9 silver badges20 bronze badges

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

Collectives™ on Stack Overflow

Implement own function

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