Computer >> Máy Tính >  >> Lập trình >> Python

Làm thế nào để tìm các cột số trong Pandas?

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

Các bước

  • Tạo dữ liệu dạng bảng hai chiều, có thể thay đổi kích thước, có khả năng không đồng nhất, df .
  • In DataFrame đầu vào, df .
  • Lập danh sách kiểu dữ liệu, tức là số, để chọn một cột.
  • Trả về một tập hợp con các cột của DataFrame dựa trên cột loại .
  • In cột có kiểu dữ liệu là 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

Đầu ra

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