Python - เลือกคอลัมน์ที่มีประเภทข้อมูลเฉพาะ

ในการเลือกคอลัมน์ที่มีประเภทข้อมูลเฉพาะ ให้ใช้ select_dtypes() วิธีการและ รวม พารามิเตอร์. ขั้นแรก ให้สร้าง DataFrame ที่มี 2 คอลัมน์ -

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

ตอนนี้ เลือก 2 คอลัมน์ที่มีประเภทข้อมูลเฉพาะที่เกี่ยวข้อง -

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

ตัวอย่าง

ต่อไปนี้เป็นรหัส -

import pandas as pd

# Create DataFrame
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"

# get the description
print(dataFrame.info())

# select columns with specific datatype
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

ผลลัพธ์

สิ่งนี้จะสร้างผลลัพธ์ต่อไปนี้ -

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:

<class 'pandas.core.frame.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([u'Student'], dtype='object')
Column 2 with int64 type = Index([u'Roll Number'], dtype='object')