python - load_iris() 得到了一个意外的关键字参数 'as_frame'

Question

我尝试将 iris 数据集导入 daraframe，但它显示以下错误。我检查了 scikit-learn 文档，其中有 as_frame 为 load_iris() 命名的参数。

我的代码:

from sklearn.datasets import load_iris
df = load_iris(as_frame=True)

错误:

TypeError                                 Traceback (most recent call last)
<ipython-input-7-1f51689afac6> in <module>
      1 from sklearn.datasets import load_iris
----> 2 df = load_iris(as_frame=True)
      3 df

TypeError: load_iris() got an unexpected keyword argument 'as_frame'

Answer 1

这可能是一个不错的选择:

from sklearn.datasets import load_iris
import pandas as pd

data = load_iris()
df = pd.DataFrame(data.data, columns=data.feature_names)
df.head()