web-dev-qa-db-fra.com

Commentaire passer plusieurs arguments à la fonction s'appliquent

J'ai une méthode appelée comptage qui prend 2 arguments. J'ai besoin d'appeler cette méthode à l'aide de la méthode apply (). Cependant, lorsque je passe les deux paramètres à la méthode apply, l'erreur suivante apparaît:

TypeError: counting () prend exactement 2 arguments (1 donné)

J'ai vu le fil suivant python pandas: appliquer une fonction avec des arguments à une série. Mise à jour et je ne souhaite pas utiliser functool.partial car je ne souhaite pas importer de classes supplémentaires pour pouvoir passer des paramètres.

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

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

Si je donne un seul paramètre, ça marche:

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

DF['new_column'] = DF['dic_column'].apply(counting)
pythonpython-3.xpandas
8
Bonson

Vous pouvez simplement utiliser un lambda:

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

D'un autre côté, il n'y a absolument rien de mal à utiliser partial ici:

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

Comme @EdChum le mentionne, si votre méthode counting cherche en fait un mot ou le remet à zéro, vous pouvez simplement utiliser la méthode pratique dict.get au lieu d'en écrire un vous-même:

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

Et un moyen non -lambda de faire ce qui précède, via le module operator:

from operator import methodcaller
count_Word = methodcaller(get, 'Word', 0)
DF['new_column'] = DF['dic_column'].apply(count_Word)
13
tzaman

La réponse acceptée est totalement parfaite. Il m'a appris des choses intéressantes sur Python. Mais juste pour le plaisir, voici plus précisément ce que nous recherchons:

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))

Merci d'avoir posté la question!

3
L0j1k