当前位置:   article > 正文

Python数据集可视化:抽取数据集的两个特征进行二维可视化、主成分分析PCA对数据集降维进行三维可视化(更好地理解维度之间的相互作用)_以iris数据集为例,对该数据集的任意两个特征,使用python画图工具,将样本显示在二

以iris数据集为例,对该数据集的任意两个特征,使用python画图工具,将样本显示在二

Python数据集可视化:抽取数据集的两个特征进行二维可视化、主成分分析PCA对数据集降维进行三维可视化(更好地理解维度之间的相互作用)

 

目录

Python数据集可视化:抽取数据集的两个特征进行二维可视化、主成分分析PCA对数据集降维进行三维可视化(更好地理解维度之间的相互作用)

#抽取数据集的两个特征进行二维可视化

#PCA对数据集降维进行三维可视化(更好地理解维度之间的相互作用)


#抽取数据集的两个特征进行二维可视化

  1. print(__doc__)
  2. # Code source: Gaël Varoquaux
  3. # Modified for documentation by Jaques Grobler
  4. # License: BSD 3 clause
  5. import matplotlib.pyplot as plt
  6. from mpl_toolkits.mplot3d import Axes3D
  7. from sklearn import datasets
  8. from sklearn.decomposition import PCA
  9. # import some data to play with
  10. iris = datasets.load_iris()
  11. X = iris.data[:, :2] # we only take the first two features.
  12. y = iris.target
  13. x_min, x_max = X[:, 0].min() - .5, X[:, 0].max() + .5
  14. y_min, y_max = X[:, 1].min() - .5, X[:, 1].max() + .5
  15. plt.figure(2, figsize=(8, 6))
  16. plt.clf()
  17. # Plot the training points
  18. plt.scatter(X[:, 0], X[:, 1], c=y, cmap=plt.cm.Set1,
  19. edgecolor='k')
  20. plt.xlabel('Sepal length')
  21. plt.ylabel('Sepal width')
  22. plt.xlim(x_min, x_max)
  23. plt.ylim(y_min, y_max)
  24. plt.xticks(())
  25. plt.yticks(())

#PCA对数据集降维进行三维可视化(更好地理解维度之间的相互作用)

  1. # To getter a better understanding of interaction of the dimensions
  2. # plot the first three PCA dimensions
  3. fig = plt.figure(1, figsize=(8, 6))
  4. ax = Axes3D(fig, elev=-150, azim=110)
  5. X_reduced = PCA(n_components=3).fit_transform(iris.data)
  6. ax.scatter(X_reduced[:, 0], X_reduced[:, 1], X_reduced[:, 2], c=y,
  7. cmap=plt.cm.Set1, edgecolor='k', s=40)
  8. ax.set_title("PCA directions for viz")
  9. ax.set_xlabel("1st eigenvector")
  10. ax.w_xaxis.set_ticklabels([])
  11. ax.set_ylabel("2nd eigenvector")
  12. ax.w_yaxis.set_ticklabels([])
  13. ax.set_zlabel("3rd eigenvector")
  14. ax.w_zaxis.set_ticklabels([])
  15. plt.show()

 

参考:sklearn

参考:The Iris Dataset

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/Gausst松鼠会/article/detail/354588
推荐阅读
相关标签
  

闽ICP备14008679号