Python разреженная матрица получить максимальные значения и индекс

Question

У меня есть разреженная матрица А (равной 10 * 3 в плотном), таких как:

print type(A) 
<class scipy.sparse.csr.csr_matrix> 

print A 
(0, 0) 0.0160478743808 
(0, 2) 0.0317314165078 
(1, 2) 0.0156596521648 
(1, 0) 0.0575683686558 
(2, 2) 0.0107481166871 
(3, 0) 0.0150580924929 
(3, 2) 0.0297743235876 
(4, 0) 0.0161931803955 
(4, 2) 0.0320187296788 
(5, 2) 0.0106034409766 
(5, 0) 0.0128109177074 
(6, 2) 0.0105766993238 
(6, 0) 0.0127786088452 
(7, 2) 0.00926522256063 
(7, 0) 0.0111941023699 

В максимальных значениях для каждого столбца:

print A.max(axis=0) 
(0, 0) 0.0575683686558 
(0, 2) 0.0320187296788 

Я хотел бы получить индекс , соответствующий значению столбца. Я знаю, что

A.getcol(i).tolist()

вернет мне список каждого столбца, который позволит мне использовать функцию argmax(), но этот способ очень медленный. Мне интересно, есть ли способ спуска?

Answer 1

Это небольшое изменение метода вы предложили в вопросе:

col_argmax = [A.getcol(i).A.argmax() for i in range(A.shape[1])] 

(Атрибут .A эквивалентно .toarray().)

Потенциально более эффективной альтернативой является

B = A.tocsc() 
col_argmax = [B.indices[B.indptr[i] + B.data[B.indptr[i]:B.indptr[i+1]].argmax()] for i in range(len(B.indptr)-1)] 

---

Любой из перечисленных выше будет работать , но я должен спросить: если ваш массив имеет форму (10, 3), почему вы используете разреженную матрицу? (10, 3) мало! Просто используйте регулярный, плотный массив numpy.

Даже если вы держите A как разреженную матрицу, наиболее эффективный способ вычисления Argmax столбцов матрицы такого размера, вероятно, просто преобразовать его в плотный массив и использовать Argmax метод:

col_argmax = A.A.argmax(axis=0)