koalas dataframe apply lambda two columns code example

Example: lambda with two columns pandas

tp['col'] = tp.apply(lambda row: row['source'] if row['target'] in ['b', 'n'] else 'x',
                     axis=1)