새싹/TIL

[핀테커스] 231005 matplot 실습

jykim23 2023. 10. 5. 16:08

케글의 아이리스 데이터로 시각화 실습

 

시각화 별거 아니라고 생각했는데 현실은 내가 별거였다.

그리고 ChatGPT의 최적화 코드를 보고 한번 더 현타.

열심히 하자.

 

import matplotlib.pyplot as plt
from sklearn.datasets import load_iris
import numpy as np

figsize = (10, 10)
iris = load_iris()
iris_X, iris_y = iris.data, iris.target
feature_names = iris.feature_names
n_feature = len(feature_names)

fig, axes = plt.subplots(4, 4, figsize=figsize)

for i, ax in enumerate(axes.flat):
    row_feat, col_feat = i // n_feature, i % n_feature

    ax.spines['top'].set_visible(False)
    ax.spines['right'].set_visible(False)

    if row_feat == 0:
        ax.set_ylabel(feature_names[col_feat], fontsize=13)
    if col_feat == n_feature-1:
        ax.set_xlabel(feature_names[row_feat], fontsize=13)

    if row_feat == col_feat:
        ax.hist(iris_X[:, row_feat], rwidth=0.9)
    else:
        ax.scatter(iris_X[:, row_feat], iris_X[:, col_feat], c=iris_y, alpha=0.5)
        ax.grid()

plt.tight_layout()
plt.show()

 

 

결과