data_analytics

import seaborn as sns
import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
from sklearn import neighbors, datasets

# importar o dataset
dataset = datasets.load_iris()

# passar para um dataframe do Pandas
data = dataset.data
target = dataset.target
target = target.reshape(target.size,1)
total_data = np.concatenate([data,target],axis=1)
total_names = np.concatenate([dataset.feature_names,["output class"]])
df = pd.DataFrame(total_data, columns=total_names)

# criar a matriz de correlação
corr = df.corr()

# criar o heatmap e mostrá-lo
sns.heatmap(corr,annot=True)
plt.show()