sparse scipy 从稀疏矩阵中选取某些指定列构成新的稀疏矩阵

sparse scipy 从稀疏矩阵中选取某些指定列构成新的稀疏矩阵
主要想法:
1.getcol(i)获得指定行
2.hstack([A,B]) 合并两个矩阵
注意:hstack返回的矩阵是以coo.matrix形式存储的。

from scipy.sparse import  hstack
def combine(row_indices,feature_matrix):   
#row_indices删除列的list
#feature_matrix初始的稀疏矩阵
#采用二分法,减少hstack操作。
    left = 0
    right = len(row_indices)-1
    mid = (left+right)/2
    matrix1 = feature_matrix.getcol(row_indices[left])
    matrix2 = feature_matrix.getcol(row_indices[right])
    if left+1>=right:
        test = hstack([matrix1,matrix2])
        return test
    while left+1

你可能感兴趣的:(sparse scipy 从稀疏矩阵中选取某些指定列构成新的稀疏矩阵)