python - 如何根据 dtype 删除 DataFrame 列

Question

我有一个 pandas 数据框，我正在尝试删除所有对象字段，以便只剩下数字。

我一直在尝试编写一个 for 循环来完成此任务，因为我可能需要使用不同的数据一遍又一遍地执行此任务。

由于某种原因我无法让它工作。以下是我到目前为止所做的事情

for cols in data:
    if data.values.type == object:
        numdata = data.drop(axis=1, inplace=True)

我得到的错误是:

AttributeError Traceback (most recent call last) in () 1 for cols in data: ----> 2 if data.values.type == object: 3 numdata = data.drop(axis=1, inplace=True)

AttributeError: 'numpy.ndarray' object has no attribute 'type'

我是一个新手，由于某种原因，我无法记住 for 循环和 if 语句逻辑。

Answer 1

您可以使用select_dtypes排除特定类型的列。

import pandas as pd

df = pd.DataFrame({'x': ['a', 'b', 'c'], 'y': [1, 2, 3], 'z': ['d', 'e', 'f']})

df = df.select_dtypes(exclude=['object'])
print(df)