Чтобы выбрать столбцы, которые имеют только числовой тип данных из Pandas DataFrame, вызовите метод DataFrame.select_dtypes() и передайте np.number или ‘number’ в качестве аргумента для параметра include. Метод DataFrame.select_dtypes() для данного аргумента возвращает подмножество этого DataFrame только с числовыми столбцами.
Синтаксис для вызова метода:
DataFrame.select_dtypes(include=None, exclude=None)
Пример 1
В этом примере мы создали DataFrame, и с помощью метода DataFrame.select_dtypes() мы получим только те столбцы, которые являются числовыми.
import pandas as pd
df = pd.DataFrame(
[['abc', 22, 22.6],
['xyz', 25, 23.2],
['pqr', 31, 30.5]],
columns=['name', 'age', 'bmi'])
result = df.select_dtypes(include='number')
print(result)
Вывод:
age bmi 0 22 22.6 1 25 23.2 2 31 30.5
Из этого DataFrame выбираются только столбцы с числовым типом данных.
В этом руководстве мы узнали, как выбирать только числовые столбцы DataFrame с помощью метода DataFrame.select_dtypes().