Python

How to Convert Sklearn Dataset to Dataframe

In this post, you will learn how to convert Sklearn.datasets to Pandas Dataframe. It will be useful to know this technique (code example) if you are comfortable working with Pandas Dataframe. You will be able to perform several operations faster with the dataframe.

Sklearn datasets class comprises of several different types of datasets including some of the following:

  • Iris
  • Breast cancer
  • Diabetes
  • Boston
  • Linnerud
  • Images

The code sample below is demonstrated with IRIS data set. Before looking into the code sample, recall that IRIS dataset when loaded has data in form of “data” and labels present as “target”.

import pandas as pd
import matplotlib.pyplot as plt
from sklearn import datasets

# Load the IRIS dataset
iris = datasets.load_iris()
X = iris.data
y = iris.target

# Create dataframe using iris.data
df = pd.DataFrame(data=iris.data, columns=["sepal_length", "sepal_width", "petal_length", "petal_width"])

# Append class / label data
df["class"] = iris.target

# Print the data and check for yourself
df.head()

Executing the above code will print the following dataframe.

Fig 1. IRIS dataset represented as Pandas dataframe

In case, you don’t want to explicitly assign column name, you could use the following commands:

# Create dataframe using iris.data
df = pd.DataFrame(data=iris.data)

# Append class / label data
df["class"] = iris.target

# Print the data and check for yourself
df.head()

Conclusion

In this post, you learned about how to convert the SKLearn dataset to Pandas DataFrame.

Ajitesh Kumar

I have been recently working in the area of Data analytics including Data Science and Machine Learning / Deep Learning. I am also passionate about different technologies including programming languages such as Java/JEE, Javascript, Python, R, Julia, etc, and technologies such as Blockchain, mobile computing, cloud-native technologies, application security, cloud computing platforms, big data, etc. I would love to connect with you on Linkedin. Check out my latest book titled as First Principles Thinking: Building winning products using first principles thinking.

Recent Posts

Agentic Reasoning Design Patterns in AI: Examples

In recent years, artificial intelligence (AI) has evolved to include more sophisticated and capable agents,…

1 month ago

LLMs for Adaptive Learning & Personalized Education

Adaptive learning helps in tailoring learning experiences to fit the unique needs of each student.…

1 month ago

Sparse Mixture of Experts (MoE) Models: Examples

With the increasing demand for more powerful machine learning (ML) systems that can handle diverse…

2 months ago

Anxiety Disorder Detection & Machine Learning Techniques

Anxiety is a common mental health condition that affects millions of people around the world.…

2 months ago

Confounder Features & Machine Learning Models: Examples

In machine learning, confounder features or variables can significantly affect the accuracy and validity of…

2 months ago

Credit Card Fraud Detection & Machine Learning

Last updated: 26 Sept, 2024 Credit card fraud detection is a major concern for credit…

2 months ago