Python machine learning-sklearning

特征值,特征向量排序

2017-09-30  本文已影响5人  阿发贝塔伽马
import numpy as np

evals=np.array([0,2,5,3,1])
evecs=np.array([[1,1],[2,3],[4,5],[7,7],[6,8]])
print evecs.shape
# 默认按升序排列
sorted_indices = np.argsort(evals)
# 取相反数 变为按降序排列
sorted_indices = np.argsort(-evals)
print sorted_indices
k=3
# 切片取特征向量列向量
topk_evecs = evecs.T[:,sorted_indices[:k]]
print topk_evecs
上一篇 下一篇

猜你喜欢

热点阅读