Python Pandas Apply, Pass Dictionary как аргумент

Question

Я хочу передать словарь в качестве дополнительного аргумента функции. Функция должна применяться к каждой строке кадра данных. Поэтому я использую «apply». Ниже у меня есть небольшой пример моей попытки:

import pandas as pd 
import numpy as np 

def fun(df_row, dict1): 
return df_row['A']*2*max(dict1['x']) 

df = pd.DataFrame(np.random.randn(6,2),columns=list('AB')) 
dict_test = {'x': [1,2,3,4], 'y': [5,6,7,8]} 
df['D'] = df.apply(fun, args = (dict_test), axis = 1) 

Я получаю следующее сообщение об ошибке: ('весело() занимает ровно 1 аргумент (3 дано)', u'occurred по индексу 0') I использовать ** dict1 для обозначения пар ключ-значение в функции «весело»

Как ни странно, если я прохожу два заданы параметры, все работает отлично

def fun(df_row, dict1, dict2): 
return df_row['A']*2*max(dict1['x']) 

df = pd.DataFrame(np.random.randn(6,2),columns=list('AB')) 
dict_test = {'x': [1,2,3,4], 'y': [5,6,7,8]} 
df['D'] = df.apply(fun, axis = 1, args = (dict_test, dict_test)) 

Answer 1

Проблема в том, что вы не передаете кортеж, (dict_test) не является кортежем, это то же самое, что и dict_test. Вам нужен кортеж с dict_test как единственный элемент, то есть (dict_test,).

df['D'] = df.apply(fun, args=(dict_test,), axis=1) 

Answer 2

с pd.DataFrame.apply документ:

Parameters 
---------- 
... 
args : tuple 
    Positional arguments to pass to function in addition to the 
    array/series 
Additional keyword arguments will be passed as keywords to the function 

Последняя строка означает, что если вы хотите распаковать dict_test передать аргументы в качестве ключевых слов к fun, вы можете сделать:

df['D'] = df.apply(fun, axis=1, **dict_test) 

Ранее вы использовали:

df['D'] = df.apply(fun, args = (dict_test), axis = 1) 

Это значит, что apply попытается позвонить в fun таким образом:

fun(df_row, dict_test) 

Но это не, как вы определили fun (вы получите разницу, когда вы знаете немного больше о позиционных и именованных аргументов).