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

后端 未结 22 2094
清酒与你
清酒与你 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:23

    Took me 2 hours to figure this out

    import numpy as np
    import pandas as pd
    from sklearn.datasets import load_iris
    
    iris = load_iris()
    ##iris.keys()
    
    
    df= pd.DataFrame(data= np.c_[iris['data'], iris['target']],
                     columns= iris['feature_names'] + ['target'])
    
    df['species'] = pd.Categorical.from_codes(iris.target, iris.target_names)
    

    Get back the species for my pandas

提交回复
热议问题