Python - Chọn cột với kiểu dữ liệu cụ thể

Để chọn cột với kiểu dữ liệu cụ thể, dùng phương thức select_dtypes() với tham số include.

Tạo DataFrame 2 cột:

dataFrame = pd.DataFrame({
    "Student": ['Jack', 'Robin', 'Ted', 'Marc', 'Scarlett', 'Kat', 'John'],
    "Roll Number": [5, 10, 3, 8, 2, 9, 6]
})

Chọn 2 cột với kiểu dữ liệu tương ứng:

column1 = dataFrame.select_dtypes(include=['object']).columns
column2 = dataFrame.select_dtypes(include=['int64']).columns

Ví dụ

import pandas as pd

dataFrame = pd.DataFrame({
    "Student": ['Jack', 'Robin', 'Ted', 'Marc', 'Scarlett', 'Kat', 'John'],
    "Roll Number": [5, 10, 3, 8, 2, 9, 6]
})
print("DataFrame ...\n", dataFrame)
print("\nInfo of the entire dataframe:\n")
print(dataFrame.info())

column1 = dataFrame.select_dtypes(include=['object']).columns
column2 = dataFrame.select_dtypes(include=['int64']).columns
print("Column 1 with object type =", column1)
print("Column 2 with int64 type =", column2)

Đầu ra

DataFrame ...
   Roll Number   Student
0            5      Jack
1           10     Robin
2            3       Ted
3            8      Marc
4            2  Scarlett
5            9       Kat
6            6      John

Info of the entire dataframe:

RangeIndex: 7 entries, 0 to 6
Data columns (total 2 columns):
Roll Number    7 non-null int64
Student        7 non-null object
dtypes: int64(1), object(1)
memory usage: 184.0+ bytes
None

Column 1 with object type = Index(['Student'], dtype='object')
Column 2 with int64 type = Index(['Roll Number'], dtype='object')