用 pandas 画散点图矩阵


用 pandas 画散点图矩阵

散点图矩阵就是将所有特征两两结合,画出他们的散点图。

用到的方法:

pd.plotting.scatter_matrix()

以鸢尾花数据集为例(在 Jupyter lab 中运行):

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

iris_dataset = load_iris()
# 查看数据集的描述
# print(iris_dataset.DESCR)
import pandas as pd

iris_df = pd.DataFrame(iris_dataset['data'],columns=iris_dataset.feature_names)
iris_df.head()

pd.plotting.scatter_matrix(iris_df,  c=iris_dataset['target'],figsize=(15,15),marker='o',alpha=0.5)
plt.savefig('matrix.png') # 保存


免责声明!

本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系本站邮箱yoyou2525@163.com删除。



 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM