在Python中,使用select_dtypes
方法篩選字符串類型的列,可以通過指定數(shù)據(jù)類型為object
來實現(xiàn)。以下是一個示例代碼:
import pandas as pd
# 創(chuàng)建一個包含不同數(shù)據(jù)類型的DataFrame
data = {'col1': [1, 2, 3, 4],
'col2': ['a', 'b', 'c', 'd'],
'col3': [True, False, True, False]}
df = pd.DataFrame(data)
# 選擇數(shù)據(jù)類型為字符串的列
str_cols = df.select_dtypes(include='object').columns
print(str_cols)
運行以上代碼,將輸出Index(['col2'], dtype='object')
,表示col2
列中包含字符串類型的數(shù)據(jù)。