Pandas: как применить функцию, которая принимает массив к столбцу?

Question

Итак, у меня есть функция, которая выполняет итерацию по массиву элементов столбца. Я не могу заставить его работать для каждого элемента, потому что он должен перебирать все элементы линейно, имея некоторый контекст with. Итак, как сделать что-то вроде этого:

df['newColumn'] = fun(df['oldColumn'])

Как это сделать в pandas?

Answer 1

Если вывод похож на массив с той же длиной, что и df, вы можете попробовать:

df['newColumn'] = pd.Series(fun(df['oldColumn']), index = df.index)

Answer 2

У вас есть два варианта. Либо создайте новую функцию fun2(), имеющую ту же функциональность, что и fun(), но только для одного элемента, и выполните

df['newColumn'] = df['oldColumn'].apply(lambda x: fun2(x))

, который будет применять fun2() к каждому элементу в столбец, или приведите df['oldColumn'] к списку, позвонив df['oldColumn'].tolist()