Як вибрати лише числові стовпці в pandas

Ви можете використовувати наступний базовий синтаксис, щоб вибрати лише числові стовпці в pandas DataFrame:

 import pandas as pd
import numpy as np

df. select_dtypes (include= np.number )

У наступному прикладі показано, як використовувати цю функцію на практиці.

Приклад: виберіть лише числові стовпці в Pandas

Припустімо, що у нас є такий фрейм даних pandas, який містить інформацію про різних баскетболістів:

 import pandas as pd

#createDataFrame
df = pd. DataFrame ({' team ': ['A', 'B', 'C', 'D', 'E', 'F', 'G', 'H'],
                   ' points ': [18, 22, 19, 14, 14, 11, 20, 28],
                   ' assists ': [5, 7, 7, 9, 12, 9, 9, 4],
                   ' rebounds ': [11, 8, 10, 6, 6, 5, 9, 12]})

#view DataFrame
print (df)

  team points assists rebounds
0 A 18 5 11
1 B 22 7 8
2 C 19 7 10
3 D 14 9 6
4 E 14 12 6
5 F 11 9 5
6 G 20 9 9
7:28 4 12

Ми можемо використовувати такий синтаксис, щоб вибрати лише числові стовпці з DataFrame:

 import numpy as np

#select only the numeric columns in the DataFrame
df. select_dtypes (include= np.number )

        points assists rebounds
0 18 5 11
1 22 7 8
2 19 7 10
3 14 9 6
4 14 12 6
5 11 9 5
6 20 9 9
7 28 4 12

Зауважте, що вибрано лише три цифрові стовпці – очки , передачі та підбирання .

Ми можемо переконатися, що ці стовпці є числовими, використовуючи функцію dtypes() для відображення типу даних кожної змінної у DataFrame:

 #display data type of each variable in DataFrame
df. dtypes

team object
int64 dots
assists int64
rebounds int64
dtype:object

З результату ми бачимо, що команда є об’єктом (тобто рядком), а очки , передачі та підбирання – це числові значення.

Зауважте, що ми також можемо використати наступний код, щоб отримати список числових стовпців DataFrame:

 #display list of numeric variables in DataFrame
df. select_dtypes (include=np. number ). columns . tolist ()

['points', 'assists', 'rebounds']

Це дозволяє нам швидко бачити імена числових змінних у DataFrame, не бачачи їхніх фактичних значень.

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

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

Як вибрати стовпці за назвою в Pandas
Як вибрати стовпці за індексом у Pandas
Як вибрати стовпці, що містять певний рядок у Pandas