เพื่อหาคอลัมน์ตัวเลขใน Pandas เราสามารถสร้าง List ของจำนวนเต็ม และนำเข้าไปในเมธอด select_dtypes() มาลองตัวอย่างดูวิธีใช้เมธอดนี้
ขั้นตอน
- สร้างข้อมูลตารางสองมิติ df
- พิมพ์ DataFrame อินพุต df
- สร้าง List ของชนิดข้อมูล คือ numerics เพื่อเลือกคอลัมน์
- คืนค่าส่วนย่อยของคอลัมน์ DataFrame ตาม dtypes ของคอลัมน์
- พิมพ์คอลัมน์ที่มีชนิดข้อมูล int
ตัวอย่าง
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)
ผลลัพธ์
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
