numpy 性能提升

a = np.array([1,2,3,4,5,1,2,2,2])
c = np.unique(a)
print(c)


对于很大的稀疏矩阵,我们不能用
a[a>0]去取大于0的元素,而应该使用np.where(a>0),得到索引[2,5,67,,100]
然后通过索引直接获取元素,这样会很快
如果要实现两个array的索引并集,就都取出来,进行hstack,然后使用np.unique


找出unique元素
posted @ 2019-03-29 13:35  yjy888  阅读(213)  评论(0编辑  收藏  举报