Cách tìm cột số trong Pandas

Để tìm cột số (numeric columns) trong Pandas, chúng ta có thể tạo danh sách các kiểu số nguyên và đưa nó vào phương thức select_dtypes(). Hãy lấy ví dụ và xem cách áp dụng phương thức này.

Các bước

• Tạo dữ liệu bảng hai chiều, df.
• In DataFrame đầu vào, df.
• Tạo danh sách kiểu dữ liệu, ví dụ numerics, để chọn cột.
• Trả về tập con các cột của DataFrame dựa trên dtypes của cột.
• In cột có kiểu dữ liệu int.

Ví dụ

import pandas as pd
df = pd.DataFrame(
    dict(
        name=['John', 'Jacob', 'Tom', 'Tim', 'Ally'],
        marks=[89, 23, 100, 56, 90],
        subjects=["Math", "Physics", "Chemistry", "Biology", "English"]
    )
)
print("Input DataFrame is:\n", df)
numerics = ['int16', 'int32', 'int64']
df = df.select_dtypes(include=numerics)
print("Numeric column in input DataFrame is:\n", df)

Kết quả

Input DataFrame is:
   name  marks subjects
0  John     89     Math
1 Jacob     23 Physics
2   Tom    100 Chemistry
3   Tim     56 Biology
4  Ally     90 English
Numeric column in input DataFrame is:
   marks
0     89
1     23
2    100
3     56
4     90