使用python生成示例（虹膜）數據集

Question

我想嘗試使用python進行一些操作，並且與R一起使用時，有一個非常方便的功能，可以獲取這樣的示例數據集

> v <- iris
> v
    Sepal.Length Sepal.Width Petal.Length Petal.Width    Species
1            5.1         3.5          1.4         0.2     setosa
2            4.9         3.0          1.4         0.2     setosa
3            4.7         3.2          1.3         0.2     setosa
4            4.6         3.1          1.5         0.2     setosa
5            5.0         3.6          1.4         0.2     setosa
6            5.4         3.9          1.7         0.4     setosa
7            4.6         3.4          1.4         0.3     setosa
8            5.0         3.4          1.5         0.2     setosa
9            4.4         2.9          1.4         0.2     setosa

numpy是否有可能生成一些示例數據集以供使用？

Answer 1

Scikit learning實際上提供了該數據集（如@sascha所述）。 您可以使用load_iris() ：

from sklearn.datasets import load_iris
data = load_iris()

您可以將其放入帶有熊貓的數據框中，以更好地復制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']
)