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

by

Manually, you can use pd.DataFrame constructor, giving a numpy array (data) and a list of the names of the columns (columns).
To have everything in one DataFrame, you can concatenate the features and the target into one numpy array with np.c_[...] (note the []):

import numpy as np
import pandas as pd
from sklearn.datasets import load_iris

# save load_iris() sklearn dataset to iris
# if you'd like to check dataset type use: type(load_iris())
# if you'd like to view list of attributes use: dir(load_iris())
iris = load_iris()

# np.c_ is the numpy concatenate function
# which is used to concat iris['data'] and iris['target'] arrays 
# for pandas column argument: concat iris['feature_names'] list
# and string list (in this case one string); you can make this anything you'd like..  
# the original dataset would probably call this ['Species']
data1 = pd.DataFrame(data= np.c_[iris['data'], iris['target']],
                     columns= iris['feature_names'] + ['target'])

More Related Contents:

  1. Why does the following code cant get value of n_fold = 1?
  2. Label encoding across multiple columns in scikit-learn
  3. How to one-hot-encode from a pandas column containing a list?
  4. Convert floats to ints in Pandas?
  5. ValueError: numpy.ndarray size changed, may indicate binary incompatibility. Expected 88 from C header, got 80 from PyObject
  6. How to one hot encode variant length features?
  7. pandas dataframe columns scaling with sklearn
  8. Efficiently create sparse pivot tables in pandas?
  9. Impute categorical missing values in scikit-learn
  10. Can You Consistently Keep Track of Column Labels Using Sklearn’s Transformer API?
  11. Custom transformer for sklearn Pipeline that alters both X and y
  12. Scikit Learn OneHotEncoder fit and transform Error: ValueError: X has different shape than during fitting
  13. LabelEncoder: TypeError: ‘>’ not supported between instances of ‘float’ and ‘str’
  14. What are the pros and cons between get_dummies (Pandas) and OneHotEncoder (Scikit-learn)?
  15. Run an OLS regression with Pandas Data Frame
  16. TfidfVectorizer in scikit-learn : ValueError: np.nan is an invalid document
  17. How to use sklearn fit_transform with pandas and return dataframe instead of numpy array?
  18. Stratified Sampling in Pandas
  19. Got continuous is not supported error in RandomForestRegressor
  20. How to change the order of DataFrame columns?
  21. pandas loc vs. iloc vs. at vs. iat?
  22. How to drop rows from pandas data frame that contains a particular string in a particular column? [duplicate]
  23. Find the end of the month of a Pandas DataFrame Series
  24. Vectorised Haversine formula with a pandas dataframe
  25. Read a small random sample from a big CSV file into a Python data frame
  26. How to concatenate two dataframes without duplicates?
  27. Most efficient way to forward-fill NaN values in numpy array
  28. How to read a list of parquet files from S3 as a pandas dataframe using pyarrow?
  29. append dictionary to data frame
  30. reading multiple files contained in a zip file with pandas

Leave a Comment