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

后端 未结 22 2132
清酒与你
清酒与你 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条回答
  •  Happy的楠姐
    2020-11-28 19:24

    I took couple of ideas from your answers and I don't know how to make it shorter :)

    import pandas as pd
    from sklearn.datasets import load_iris
    iris = load_iris()
    df = pd.DataFrame(iris.data, columns=iris['feature_names'])
    df['target'] = iris['target']
    

    This gives a Pandas DataFrame with feature_names plus target as columns and RangeIndex(start=0, stop=len(df), step=1). I would like to have a shorter code where I can have 'target' added directly.

提交回复
热议问题