Как передать несколько аргументов функции apply

Question

У меня есть метод под названием counting, который принимает 2 аргумента. Мне нужно вызвать этот метод, используя метод apply(). Однако, когда я передаю два параметра для метода применения он дает следующее сообщение об ошибке:

TypeError: counting() takes exactly 2 arguments (1 given)

я видел следующую нить python pandas: apply a function with arguments to a series. Update, и я не хочу использовать functool.partial как я не хочу, чтобы импортировать дополнительные классы, чтобы иметь возможность передавать параметры.

def counting(dic, strWord): 
    if strWord in dic: 
     return dic[strWord] 
    else: 
     return 0 

DF['new_column'] = DF['dic_column'].apply(counting, 'word') 

Если я дам один параметр, он работает:

def awesome_count(dic): 
    if strWord in dic: 
     return dic[strWord] 
    else: 
     return 0 

DF['new_column'] = DF['dic_column'].apply(counting) 

Answer 1

Вы можете просто использовать lambda:

DF['new_column'] = DF['dic_column'].apply(lambda dic: counting(dic, 'word')) 

С другой стороны, нет абсолютно ничего плохого в использовании partial здесь:

from functools import partial 
count_word = partial(counting, strWord='word') 
DF['new_column'] = DF['dic_column'].apply(count_word) 

Как отмечает @EdChum, если ваш метод counting на самом деле просто глядя слово или недобросовестный его к нулю, вы можете просто использовать удобный dict.get метод вместо себя написанием:

DF['new_column'] = DF['dic_column'].apply(lambda dic: dic.get('word', 0)) 

И в не- lambda способ сделать выше, с помощью operator модуля:

from operator import methodcaller 
count_word = methodcaller(get, 'word', 0) 
DF['new_column'] = DF['dic_column'].apply(count_word) 

Answer 2

Принятого ответ абсолютно совершенный. Преподавал мне некоторые интересные вещи о Python. Но только для удовольствия, а точнее, то, что мы ищем:

selected_words = ['awesome', 'great', 'fantastic', 'amazing', 'love', 'horrible', 'bad', 'terrible', 'awful', 'wow', 'hate'] 
for this_word in selected_words: 
    products[this_word] = products['word_count'].apply(lambda dic: dic.get(this_word, 0)) 

Спасибо за сообщение!