溫馨提示×

Python中怎么用select_dtypes篩選字符串

小億
89
2024-06-11 15:29:26
欄目: 編程語言

在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ù)。

0