Dataset
wget https://archive.ics.uci.edu/ml/machine-learning-databases/iris/iris.data
Code
import numpy as np
import pandas as pd
from sklearn.preprocessing import OneHotEncoder
columns = ['Sepal length', 'Sepal width', 'Petal length', 'Petal width', 'Class_labels']
df = pd.read_csv('./datasets/iris.data', names=columns)
print(df)
encoder = OneHotEncoder(handle_unknown='ignore')
encoder_df = pd.DataFrame(encoder.fit_transform(df[['Class_labels']]).toarray())
print(encoder_df)
final_df = df.join(encoder_df)
print(final_df)
final_df.drop('Class_labels', axis=1, inplace=True)
print(final_df)
final_df.to_csv("", index=0, header=0)