Pandas: удалить определенный столбец при импорте файла csv

Вы можете использовать следующий базовый синтаксис для удаления определенного столбца при импорте файла CSV в DataFrame pandas:

 df = pd. read_csv (' basketball_data.csv ', usecols= lambda x:x != ' rebounds ')

В этом конкретном примере каждый столбец из CSV-файла с именем Basketball_data.csv будет считан в DataFrame pandas, за исключением столбца с именем rebounds .

В следующем примере показано, как использовать этот синтаксис на практике.

Пример. Удаление определенного столбца при импорте файла CSV в Pandas.

Допустим, у нас есть следующий CSV-файл с именем Basketball_data.csv :

Мы можем использовать следующий синтаксис для импорта CSV-файла в pandas и удаления столбца под названием «Bounces» во время импорта:

 import pandas as pd

#import all columns except 'rebounds' into DataFrame
df = pd. read_csv (' basketball_data.csv ', usecols= lambda x:x != ' rebounds ')

#view resulting DataFrame
print (df)

  team points
0 to 22
1 B 14
2 C 29
3 D 30

Обратите внимание, что столбец отказов был удален, когда мы импортировали файл CSV в pandas.

Если вы хотите удалить несколько столбцов во время импорта, вы можете использовать оператор not in следующим образом:

 import pandas as pd

#import all columns except 'team' and 'rebounds' into DataFrame
df=pd. read_csv (' basketball_data.csv ', usecols= lambda x:x not in [' team ', ' rebounds '])

#view resulting DataFrame
print (df)

   points
0 22
1 14
2 29
3 30

Обратите внимание, что столбцы «Команда» и «Подборы» были удалены, когда мы импортировали файл CSV в pandas.

Обратите внимание, что вы можете включить в список столько имен столбцов, сколько хотите, после оператора not in , чтобы удалить столько столбцов, сколько хотите, при импорте файла CSV.

Дополнительные ресурсы

В следующих руководствах объясняется, как выполнять другие распространенные задачи на Python:

Pandas: как пропускать строки при чтении файла CSV
Pandas: как добавить данные в существующий файл CSV
Pandas: как использовать read_csv с аргументом usecols