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

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

TypeError: counting () принимает ровно 2 аргумента (1 дано)

Я видел следующий поток Python pandas: применить функцию с аргументами к серии. Update и я не хочу использовать functool.частично, как и я. не хотите импортировать дополнительные классы, чтобы иметь возможность передавать параметры.

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)
8
pandas python python-3.x

2 ответов

вы можете просто использовать 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)
13

принятый ответ совершенно идеален. Научил меня некоторым интересным вещам о 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))

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

3