How to convert a Scikit-learn dataset to a Pandas dataset?

后端 未结 22 2095
清酒与你
清酒与你 2020-11-28 19:10

How do I convert data from a Scikit-learn Bunch object to a Pandas DataFrame?

from sklearn.datasets import load_iris
import pandas as pd
data = load_iris()
p         


        
22条回答
  •  温柔的废话
    2020-11-28 19:42

    from sklearn.datasets import load_iris
    import pandas as pd
    
    iris_dataset = load_iris()
    
    datasets = pd.DataFrame(iris_dataset['data'], columns = 
               iris_dataset['feature_names'])
    target_val = pd.Series(iris_dataset['target'], name = 
                'target_values')
    
    species = []
    for val in target_val:
        if val == 0:
            species.append('iris-setosa')
        if val == 1:
            species.append('iris-versicolor')
        if val == 2:
            species.append('iris-virginica')
    species = pd.Series(species)
    
    datasets['target'] = target_val
    datasets['target_name'] = species
    datasets.head()
    

提交回复
热议问题