Tạo cột mới trong Panda bằng cách sử dụng hàm lambda trên hai cột hiện có

Question

Tôi có thể thêm cột mới trong Panda bằng cách xác định hàm người dùng và sau đó sử dụng áp dụng. Tuy nhiên, tôi muốn thực hiện việc này bằng cách sử dụng lambda; có cách nào không?

Ví dụ: df có hai cột a và b. Tôi muốn tạo cột mới c bằng chiều dài dài nhất giữa a và b.

Một số điều như:

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

Một cách tiếp cận:

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

Bạn có thể sử dụng chức năng map và chọn bởi chức năng 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 

giải pháp tiếp theo là với chức năng apply với sự nổi tiếng 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)