Início / coding / Perguntas / 79225412
Accepted
Paul Jurczak
Asked: 2024-11-26 13:22:39 +0800 CST

Por que essa operação da coluna Pandas DataFrame falha?

  • 772

Este script funciona bem com Python 3.11 e Pandas 2.2:

import pandas as pd

df = pd.read_csv(f'test.csv', comment='#')
df['x1'] = df['x1']*8
# df['y1'] = df['y1']*8
print(df)

e impressões:

   x1   y1
0   0    0
1  16    6
2  32   12

mas falha quando eu descomento df['y1'] = df['y1']*8e produz KeyError: 'y1'. Por que isso? 'y1'é uma chave válida. Aqui está o test.csvarquivo:

# comment 
x1, y1
0, 0
2, 6
4, 12
python

2 respostas

  1. Isso não é um CSV válido. É verdade que o CSV é definido de forma imprecisa, mas espaços em branco gratuitos são um problema. Tente olhar os nomes das colunas:

    >>> df.columns
Index(['x1', ' y1'], dtype='object')

    Pandas presumiu que você queria que esses espaços fizessem parte do nome da coluna. Pandas fez um bom trabalho em adivinhar os inteiros, mas então o inteiro nativo do python entende como fazer isso para

    >>> int('   4')
4

    Você pode remover o espaço em branco após ler o csv

    import pandas as pd

df = pd.read_csv(f'test.csv', comment='#')
df.columns = df.columns.str.strip()
df['x1'] = df['x1']*8
df['y1'] = df['y1']*8
print(df)

    E se os dados da coluna também tiverem o mesmo problema de espaço (talvez você tenha uma coluna de strings), você pode fazer a mesma coisa lá.

    • 1
  2. Best Answer

    Leia seu arquivo com a skipinitialspace=Trueopção deread_csv

    df = pd.read_csv(csv, skipinitialspace=True)

    Colunas:

    ['x1', 'y1']
    • 1

relate perguntas