用python生成示例（Iris）数据集

Question

我想尝试使用 python 进行一些操作，并且使用R 后，有一个非常方便的功能可以获取这样的示例数据集> v <- iris> v    Sepal.Length Sepal.Width Petal.Length Petal.Width    Species1            5.1         3.5          1.4         0.2     setosa2            4.9         3.0          1.4         0.2     setosa3            4.7         3.2          1.3         0.2     setosa4            4.6         3.1          1.5         0.2     setosa5            5.0         3.6          1.4         0.2     setosa6            5.4         3.9          1.7         0.4     setosa7            4.6         3.4          1.4         0.3     setosa8            5.0         3.4          1.5         0.2     setosa9            4.4         2.9          1.4         0.2     setosanumpy 是否可以生成一些示例数据集来玩？

Answer 1

Scikit learn 实际上提供了该数据集（如@sascha 所述）。您可以使用load_iris()：

from sklearn.datasets import load_iris

data = load_iris()

您可以使用 Pandas 将其制作成数据框，以更好地复制 R：

import pandas as pd

species = [data['target_names'][x] for x in data['target']]

pd.DataFrame(

    np.column_stack([data['data'], species]),

    columns=data['feature_names'] + ['species']

)