Как найти размер матрицы CSR в Python

Question

У меня есть tfidf матрица типа scipy.sparse.csr.csr_matrix

Как мне найти количество строк и количество столбцов в Python 3?

Я попытался преобразовать его в плотный и затем найти длину строки, как это

len(tfidf_matrix.todense()) но выдает 1, что неверно.

Answer 1

Вы ищете функцию matrix.get_shape()

https://docs.scipy.org/doc/scipy-0.14.0/reference/generated/scipy.sparse.csr_matrix.get_shape.html#scipy.sparse.csr_matrix.get_shape