Início / coding / Perguntas / 79554434
Accepted
GH KIM
Asked: 2025-04-04 11:57:33 +0800 CST

Como substituir loops aninhados por apply from dataframe em python

  • 772

Tenho um dataframe e uma lista abaixo.

import pandas as pd

my_df = pd.DataFrame({'fruits': ['apple', 'banana', 'cherry', 'durian'], 
                      'check': [False, False, False, False]})
my_list = ['pp', 'ana', 'ra', 'cj', 'up', 'down', 'pri']

>>> my_df
   fruits     check
0     apple     False
1    banana     False
2    cherry     False
3    durian     False

Posso criar um resultado com loops for aninhados.

for fruit in my_df['fruits']:
    for v in my_list:
        if v in fruit:
            my_df.loc[my_df['fruits']==fruit, 'check'] = True

>>> my_df
   fruits     check
0     apple     True
1    banana     True
2    cherry     False
3    durian     False

Eu tentei abaixo.

my_df['fruits'].apply(lambda x: True for i in my_list if i in x)

mas, cuspiuType Error: 'generator' object is not cllable

Quero remover loops for aninhados e substituí-los pela função apply. Como posso fazer isso?

python

1 respostas

  1. Best Answer

    Esta linha deve ser corrigida.

    my_df['fruits'].apply(lambda x: True for i in my_list if i in x)

    A função lambda retorna um gerador, não um booleano ( True/ False value). Por outro lado, o apply()método espera um valor de retorno para cada linha do DataFrame, e esse valor de retorno deve ser um booleano (ou qualquer outro tipo), não um objeto gerador. Além disso, falta any()verificar se há correspondências.

    A linha corrigida é:

    my_df['check'] = my_df['fruits'].apply(lambda x: any(v in x for v in my_list))

    Saída:

    insira a descrição da imagem aqui

    Fontes:

    https://docs.python.org/3/tutorial/controlflow.html#lambda-expressions

    https://docs.python.org/3/tutorial/classes.html#generator-expressions

    https://docs.python.org/3/library/functions.html#any

    • 1

relate perguntas