프로그래밍/Python
[python] pair plot
홍반장水_
2024. 12. 20. 17:40
반응형
python pair plot
데이터 세트의 숫자 변수 간의 쌍 관계를 보여주는 산점도 그리드입니다. 일반적으로 데이터 분포와 변수 간의 관계를 이해하는 데 사용됩니다.
import seaborn as sns
import matplotlib.pyplot as plt
import pandas as pd
import sklearn
print(sklearn.__version__)
# Sample dataset (Iris dataset)
from sklearn.datasets import load_iris
iris = load_iris()
df = pd.DataFrame(iris.data, columns=iris.feature_names)
df['species'] = [iris.target_names[i] for i in iris.target]
# Create a pair plot
sns.set(style="ticks")
pairplot = sns.pairplot(df, hue="species", diag_kind="kde")
# Save the pair plot as an image
output_file = "pair_plot.png"
pairplot.savefig(output_file)
print(f"Pair plot saved as {output_file}")
# Show the pair plot
plt.show()
반응형