Search code examples
pythonpandasdataframeconcatenation

How to Concat 2 columns in single column with column value check

I want to concat two column from data frame where Column1 not equals to ANY:

DataFrame :

   COLUMN1 | COLUMN2
0     A    |   FOO
1     B    |   BAR  
2    ANY   |   FOO
3    ANY   |   BAR
4     C    |   FOO

as a result I want dataframe as follows

   COLUMN1 | COLUMN2
0     A    |  FOO_A
1     B    |  BAR_B
2    ANY   |  FOO
3    ANY   |  BAR  
4     C    |  FOO_C

ANY is variable, could represent Null, EmptyString, String, Number.

Thanks.

Solution

  • You can do 

    df['COLUMN2']=df.apply(lambda row:row['COLUMN2']+'_'+row['COLUMN1'] if row['COLUMN1']!='ANY' else row['COLUMN2'],axis=1)