Tworzenie nowej kolumny w Panda przy użyciu funkcji lambda na dwóch istniejących kolumnach

Question

Jestem w stanie dodać nową kolumnę w Panda, definiując funkcję użytkownika, a następnie za pomocą przycisku Zastosuj. Jednak chcę to zrobić za pomocą lambda; czy jest jakiś sposób obejścia?

Na przykład df ma dwie kolumny: a i b. Chcę utworzyć nową kolumnę c, która jest równa najdłuższej długości między a i b.

Niektóre rzeczy jak:

df['c'] = df.apply(lambda x, len(df['a']) if len(df['a']) > len(df['b']) or len(df['b'])) 

Jedno podejście:

df = pd.DataFrame({'a':['dfg','f','fff','fgrf','fghj'], 'b' : ['sd','dfg','edr','df','fghjky']}) 

df['c'] = df.apply(lambda x: max([len(x) for x in [df['a'], df['b']]])) 
print df 
     a  b c 
0 dfg  sd NaN 
1  f  dfg NaN 
2 fff  edr NaN 
3 fgrf  df NaN 
4 fghj fghjky NaN 

Answer 1

Można użyć funkcji map i wybierz według funkcji np.wheremore info

print df 
#  a  b 
#0 aaa rrrr 
#1 bb  k 
#2 ccc  e 
#condition if condition is True then len column a else column b 
df['c'] = np.where(df['a'].map(len) > df['b'].map(len), df['a'].map(len), df['b'].map(len)) 
print df 
#  a  b c 
#0 aaa rrrr 4 
#1 bb  k 2 
#2 ccc  e 3 

Następny rozwiązanie jest z funkcją apply z parame ter axis=1:

axis = 1 or ‘columns’: apply function to each row

df['c'] = df.apply(lambda x: max(len(x['a']), len(x['b'])), axis=1)