在Python中,使用select_dtypes
方法筛选字符串类型的列,可以通过指定数据类型为object
来实现。以下是一个示例代码:
import pandas as pd
# 创建一个包含不同数据类型的DataFrame
data = {'col1': [1, 2, 3, 4],
'col2': ['a', 'b', 'c', 'd'],
'col3': [True, False, True, False]}
df = pd.DataFrame(data)
# 选择数据类型为字符串的列
str_cols = df.select_dtypes(include='object').columns
print(str_cols)
运行以上代码,将输出Index(['col2'], dtype='object')
,表示col2
列中包含字符串类型的数据。