我们现在将绘制数据集并选择可以使用滑块更改其值的任意向量。我们将数据集投影到向量跨越的线上(由滑块值定义)。我们使用选择的向量转换数据集并计算结果方差。导致最大方差的点(滑块调整)为我们提供了第一个主成分。此外,我们根据我们为向量选择的值绘制“方差曲面”。总而言之,通过改变线的方向,我们希望找到这条线,当数据集投影到这条线上时,这条线会导致最大的方差。这条线是 1. 主成分。如果我们选择平行于 x 和 y 轴的线,我们只需切断另一个轴。
import numpy as np
import pandas as pd
from ipywidgets import interact 、interactive 、fixed 、interact_manual
import matplotlib.pyplot as plt
from matplotlib import style
from mpl_toolkits.mplot3d import Axes3D
style 。使用(‘五三十八’ )
定义 f ( x , y ):
数据 = np 。阵列([[ 2.5 ,0.5 ,2.2 ,1.9 ,3.1 ,2.3 ,2 ,1 ,1.5 ,1.1 ],[ 2.4 ,0.7 ,2.9 ,2.2 ,3.0 ,2.7 ,1.6 ,1.1 ,1.6 ,0.9 ])
的数据[ 0 ] = 数据[ 0 ] - np. 均值(数据[ 0 ])
数据[ 1 ] = 数据[ 1 ] - np 。平均值(数据[ 1 ])
# 创建轴
无花果 = plt 。图( figsize = ( 10 , 10 ))
ax0 = fig . add_subplot ( 121 )
ax0 。set_aspect ( ‘equal’ )
ax0 。set_ylim ( - 2 , 2 )
ax0 。set_xlim ( - 2 , 2 )
ax0 。set_title ( ‘搜索主成分’ , fontsize =14 )
ax0 。set_xlabel ( ‘PC x value’ , fontsize = 10 )
ax0 。set_ylabel ( ‘PC y 值’ , fontsize = 10 )
#vec = np.array([0.6778734,0.73517866])
vec = np . 数组([ x , y ])
轴0 。分散(数据[ 0 ],数据[ 1 ])
ax0 。情节(NP 。linspace (分钟(数据[ 0 ]),最大(数据[ 0 ])),(VEC [ 1 ] / VEC [ 0 ])* NP 。linspace (分钟(数据[ 0 ]),最大(数据[0 ])), linewidth = 1.5 , color = “black” , linestyle = “–” )
b_on_vec_list = [[],[]]
for i in range ( len ( data [ 0 ])):
a = vec
b = np 。数组([ data [ 0 ][ i ], data [ 1 ][ i ]])
b_on_a = ( np . dot ( a , b ) / np . dot ( a , a )) *a
b_on_vec_list [ 0 ] 。追加( b_on_a [ 0 ])
b_on_vec_list [ 1 ] 。追加( b_on_a [ 1 ])
ax0 。分散( b_on_a [ 0 ], b_on_a [ 1 ], color = ‘red’ )
ax0 。绘图([ b_on_a [ 0 ], b [ 0 ]],[ b_on_a [ 1], b [ 1 ]], “r–” , linewidth = 1 )
ax1 = 图。add_subplot ( 122 ,投影= ‘3d’ )
ax1 。set_aspect ( ‘equal’ )
ax1 。set_ylim ( 0 , 1 )
ax1 。set_xlim ( 0 , 1 )
ax1 。set_title ( ‘Varicane 相对于 1. PC’ , fontsize = 14 )
ax1 . set_xlabel ( ‘PC x value’ , fontsize= 10 )
ax1 。set_ylabel ( ‘PC y value’ , fontsize = 10 )
ax1 。set_zlabel (‘方差’ ,fontsize = 10 )
#变换数据
e_vec = (1 / NP 。SQRT (NP 。点(VEC ,VEC 。Ť )))* VEC
data_trans = NP 。点(数据。Ť ,e_vec )
# 绘制数据
ax0 。scatter ( data_trans , np . zeros_like ( data_trans ), c = ‘None’ , edgecolor = ‘black’ )
# 绘制扭曲线
ax0 。情节(NP 。linspace (分钟(data_trans ),最大值(data_trans ),10 ),NP 。zeros_like (data_trans ),线型= ’ - ',颜色= ‘灰色’ ,线宽= 1.5 )
#剧情圆圈
为 我 在 范围(len个(data_trans )):
AX0 。add_artist ( plt . Circle (( 0 , 0 ), data_trans [ i ], linewidth = 0.5 , linestyle = ‘dashed’ , color = ‘grey’ , fill = False ))
# 计算方差
ax0 。文本(0 ,- 1.4 ,'方差= {0} ’ 。格式(STR (NP 。圆(NP 。VAR (data_trans ),3 ))),字体大小= 20 )
# 绘制关于主成分向量的方差