请教一个根据一定条件增加一列的问题

请教一个问题
column1    column2    column3
(A, 20)      (乙, 30)     (B, 50)
(D, 20)      (C, 70)      (E, 50)
(丁, 80)     (甲, 100)     (B, 50)

如上面所示，第一行里50最大，就选B；第二行里70最大，就选C；最三行里100最大，就选甲.
所以增加一列，第一行是B，第二行是C，第三行是甲

import pandas as pd
df = pd.DataFrame([[('A', 20), ('乙', 30), ('B', 50)],
                [('D', 20), ('C', 70), ('E', 50)],
                [('丁', 80), ('甲', 100), ('B', 50)]])

def func_a(s):
    return sorted(s.values.tolist(), key=lambda x: x[1], reverse=True)[0][0]

df['add'] = df.apply(lambda x: func_a(x), axis=1)


0        1        2        add
0        (A, 20)        (乙, 30)        (B, 50)        B
1        (D, 20)        (C, 70)        (E, 50)        C
2        (丁, 80)        (甲, 100)        (B, 50)        甲

谢谢~~~