Início / coding / Perguntas / 78336767
Accepted
iBeMeltin
Asked: 2024-04-17 02:50:15 +0800 CST

Retire todos os valores das colunas, mas verifique se existe alguma tupla

  • 772

Tenho vários dataframes que usam as mesmas funções para retirar todos os valores. No entanto, alguns dos valores são tuplas e ocorre um erro sempre que encontro a remoção de uma tupla.

Minha função fica assim:

for col in df.columns:
        if df[col].dtype == 'object':   
            df[col] = df[col].apply(lambda x: str(x).strip())

Tenho outra função para um dataframe específico que sempre terá tuplas:

for col in df.columns:
        if df[col].dtype == 'object':   
            df[col] = df[col].apply(lambda x: str(x).strip())

    df['Commodity'] = [(x.strip() for x in ls) for ls in df['Commodity'].values]
    df['Commodity'] = df['Commodity'].apply(sorted)

A segunda função sempre gerará um erro quando chegar à coluna Commodity porque não manipulará as tuplas. Como posso lidar com esse caso específico na primeira função?

python

1 respostas

  1. Best Answer

    Que tal algo como:

    for col in df.columns:
        if df[col].dtype == 'object':   
            df[col] = df[col].apply(lambda x: str(x).strip() if isinstance(x, str)
                                              else (s.strip() for s in x))

    Ou com uma função recursiva personalizada:

    def cust_strip(x):
    if isinstance(x, str):
        return x.strip()
    if isinstance(x, tuple):
        return tuple(cust_strip(item) for item in x)
    return x

for col in df.columns:
        if df[col].dtype == 'object':   
            df[col] = df[col].apply(cust_strip)

    Exemplo:

    # input
df = pd.DataFrame({'str': [' abc ', 'def '],
                   'tuple': [(' ghi', ' jkl '), ('mno ',)]})

# output:
out = pd.DataFrame({'str': ['abc', 'def'],
                    'tuple': [('ghi', 'jkl'), ('mno',)]})
    • 1

relate perguntas