load_iris() 收到意外的关键字参数“as_frame”

Question

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

我的代码：

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()