温馨提示×

Python中怎么用select_dtypes筛选字符串

小亿
86
2024-06-11 15:29:26
栏目: 编程语言

在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列中包含字符串类型的数据。

0