Pandas: як використовувати read_csv з аргументом usecols

Ви можете використовувати аргумент usecols у функції read_csv() , щоб читати певні стовпці з файлу CSV у DataFrame pandas.

Існує два поширених способи використання цього аргументу:

Спосіб 1. Використовуйте usecols із назвами стовпців

 df = pd. read_csv (' my_data.csv ', usecols=[' this_column ', ' that_column '])

Спосіб 2. Використовуйте usecols із позиціями стовпців

 df = pd. read_csv ( ' my_data.csv ', usecols=[ 0,2 ])

У наведених нижче прикладах показано, як використовувати кожен метод на практиці з таким файлом CSV під назвою basketball_data.csv :

Приклад 1: використання usecols із назвами стовпців

Ми можемо використати такий код, щоб імпортувати файл CSV і використовувати лише стовпці під назвою «команда» та «підбирання»:

 import pandas as pd

#import DataFrame and only use 'team' and 'rebounds' columns
df = pd. read_csv (' basketball_data.csv ', usecols=[' team ',' rebounds '])

#view DataFrame
print (df)

   team rebounds
0 to 10
1 B 9
2 C 6
3 D 2

Зауважте, що було імпортовано лише стовпці команди та підбирань, оскільки це були імена стовпців, які ми вказали в аргументі usecols .

Приклад 2: використання usecols із позиціями стовпців

Ми можемо використовувати наступний код для імпорту файлу CSV і використовувати лише стовпці в позиціях індексу 0 і 2:

 import pandas as pd

#import DataFrame and only use columns in index positions 0 and 2
df = pd. read_csv (' basketball_data.csv ' , usecols=[ 0,2 ])

#view DataFrame
print (df)

   team rebounds
0 to 10
1 B 9
2 C 6
3 D 2

Зауважте, що було імпортовано лише стовпці команди та підбирань, оскільки це були стовпці в індексних позиціях 0 і 2, які є значеннями, які ми вказали в аргументі usecols .

Примітка : перший стовпець файлу CSV має позицію індексу 0.

Додаткові ресурси

У наступних посібниках пояснюється, як виконувати інші типові завдання в Python:

Pandas: як пропускати рядки під час читання файлу CSV
Pandas: Як читати файли Excel
Pandas: як експортувати DataFrame в Excel