Pandas: як вибрати стовпці за типом даних
Ви можете використовувати такі методи, щоб вибрати стовпці в pandas DataFrame, які відповідають певному типу даних:
Спосіб 1: Виберіть стовпці, що відповідають певному типу даних
#select all columns that have an int or float data type
df. select_dtypes (include=[' int ', ' float '])
Спосіб 2. Виберіть різні стовпці певного типу даних
#select all columns that don't have a bool or object data type
df. select_dtypes (exclude=[' bool ', ' object '])
У наведених нижче прикладах показано, як використовувати кожен метод із такими pandas DataFrame:
import pandas as pd #createDataFrame df = pd. DataFrame ({' team ': ['A', 'B', 'C', 'D', 'E', 'F'], ' points ': [18, 22, 19, 14, 14, 11], ' assists ': [5, 7, 7, 9, 12, 9], ' minutes ': [10.1, 12.0, 9.0, 8.0, 8.4, 7.5], ' all_star ': [True, False, False, True, True, True]}) #view DataFrame print (df) team points assists minutes all_star 0 A 18 5 10.1 True 1 B 22 7 12.0 False 2 C 19 7 9.0 False 3 D 14 9 8.0 True 4 E 14 12 8.4 True 5 F 11 9 7.5 True
Приклад 1: Виберіть стовпці, що відповідають певному типу даних
Ми можемо використати наступний код, щоб вибрати всі стовпці в DataFrame, які мають тип даних, рівний int або float :
#select all columns that have an int or float data type
df. select_dtypes (include=[' int ', ' float '])
minute assist points
0 18 5 10.1
1 22 7 12.0
2 19 7 9.0
3 14 9 8.0
4 14 12 8.4
5 11 9 7.5
Зауважте, що вибираються лише стовпці з типом даних int або float .
Приклад 2: Виберіть різні стовпці певного типу даних
Ми можемо використати наступний код, щоб вибрати всі стовпці в DataFrame, які не мають типу даних, рівного bool або object :
#select all columns that don't have a bool or object data type
df. select_dtypes (exclude=[' bool ', ' object '])
minute assist points
0 18 5 10.1
1 22 7 12.0
2 19 7 9.0
3 14 9 8.0
4 14 12 8.4
5 11 9 7.5
Зауважте, що вибираються лише стовпці, тип даних яких не дорівнює bool або object .
Також зауважте, що для відображення типу даних кожного стовпця в DataFrame можна використовувати такий синтаксис:
#display data type of all columns
df. dtypes
team object
int64 dots
assists int64
minutes float64
all_star bool
dtype:object
Додаткові ресурси
У наступних посібниках пояснюється, як виконувати інші типові операції в pandas:
Pandas: як перевірити тип усіх стовпців у DataFrame
Pandas: отримати індекс рядків, стовпець яких відповідає значенню
Pandas: Як встановити стовпець як індекс