Pandas dataframe применяет функцию ко всей колонке

Question

У меня есть некоторые функции, которые действуют в списке и возвращают список. Я хотел бы создать столбец в Dataframe Pandas таким образом, чтобы новый столбец был списком, возвращаемым одной из функций, действующих на какой-либо другой столбец данных.

В питона-как псевдокод:

def function(parameter, list): 
    ... 
    return output_list 

df['New Column'] = function(parameter, df['Old Column']) 

Я пробовал разные варианты, включая то, как приведенный выше код, используя метод .Не() и другие ... не имели успеха.

Есть ли способ сделать это? Спасибо!

EDIT: См. Ответ Брайана Пендлтона для решения. Столбцы в фрейме данных представляют собой объекты серии панды. Просто нужно создать серию из нужного списка.

df['New_Column'] = pd.Series(data=function(parameter,list)) 

Answer 1

Если вы отправляете df["old column"] функции, то вы отправляете pandas.Series объект. Почему бы просто не работать с этой серией и возвращать новую серию той же формы. Затем вы можете просто использовать свое назначение в новом столбце, как есть.

Answer 2

Я думаю, что вы хотите:

• применить функцию, которая вернет список
• сохранить результат в DataFrame
• разделить список на другую колонку

Вот пример.

def my_funct(parameter): 
return (1,2,3) + parameter 

df = pd.DataFrame(np.random.randint(low=1, high=10, size=3), columns=['my_funct']) 
#df['New Column'] = function(parameter, df['Old Column']) 
df['my_funct'] = df['my_funct'].apply(lambda x: my_funct(x)) 

# After the function call the resulting lists are stored in one column 

     my_funct 
0  [5, 6, 7] 
1 [10, 11, 12] 
2  [6, 7, 8] 

# Here is how to split the list into several columns 

df = df['my_funct'].apply(pd.Series) 

    0 1 2 
0 5 6 7 
1 10 11 12 
2 6 7 8