Create function from Logic for Dataframe

Question

I have Df like this

      Name     PAN
 0    x     BBDFW7894Q 
 1    s         
 2    A     QWE7892E

i want df like this After validation PAN number

     Name     PAN           PAN_Status
 0    x     BBDFW7894Q      Valid PAN number
 1    s                     PAN is not present
 2    A     QWE7892E        Invalid PAN number

i trying with below logic

df["PAN_Status"] = ["Valid PAN number" if re.match(r'^[A-Z]{5}[0-9]{4}[A-Z]$',ele)\
                    else ("PAN is not present"if ele==' ' else "Not Valid PAN") for ele in

but i want a Function for this

jezrael · Accepted Answer · 2021-05-21 05:49:03Z

1

Use numpy.select with Series.str.match and for second condition is tested missing values or emty strings:

m1 = df["PAN"].str.match(r'^[A-Z]{5}[0-9]{4}[A-Z]$', na=False)
m2 = df["PAN"].str.strip().eq('') | df['PAN'].isna()

df["PAN_Status"] = np.select([m1, m2], 
                             ['Valid PAN number','PAN is not present'], 
                             'Not Valid PAN')

Solution with function:

def test(ele):
    if (ele.strip() =='') or pd.isna(ele):
        return "PAN is not present"
    elif re.match(r'^[A-Z]{5}[0-9]{4}[A-Z]$',ele):
        return "Valid PAN number"
    else:
        return "Not Valid PAN"

df["PAN_Status"] = df["PAN"].apply(test)
print (df)
  Name         PAN          PAN_Status
0    x  BBDFW7894Q    Valid PAN number
1    s         NaN  PAN is not present
2    A    QWE7892E       Not Valid PAN

edited May 21, 2021 at 5:49

answered May 21, 2021 at 5:24

jezrael

868k103 gold badges1.4k silver badges1.3k bronze badges

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

1 Comment

user14784453 Over a year ago

sir you are a always right but ,I want a function like 'def (): code return'

Collectives™ on Stack Overflow

Create function from Logic for Dataframe

1 Answer 1

1 Comment

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

1 Comment

Your Answer

Sign up or log in

Post as a guest

Related