モジュールsparseについて

Question

以下を実行すると答えが出るのですが、
C:\Python27\lib\site-packages\scipy\sparse\linalg\dsolve\linsolve.py:87: SparseE
fficiencyWarning: spsolve requires A be CSC or CSR matrix format
warn('spsolve requires A be CSC or CSR matrix format', SparseEfficiencyWarning
)
というメッセージが出てくるので、消すにはどうすればよいか教えてください。

import numpy
from scipy.sparse import lil_matrix
from scipy.sparse.linalg import spsolve

A = lil_matrix((4, 4))
A[0, 0] = 4
A[0, 2] = -3
A[1, 1] = 1
A[2, 2] = 1
A[3, 3] = 1
v = numpy.array([ 1, 1, 1, 1])
print spsolve(A, v)

Answer 1

ベストアンサーを参考にさせていただいた後、
少し発見がありましたので記しておきます。

Aがlil_matrixのときは**.tocsr()**を使用すると解決しますが、
Aがlistのときは上手くいきません。

そこでコードが少し長くなりますが、sparse.csr_matrixを使うと、
Aがlil_matrixのときでもlistのときでも上記のメッセージが
表示されることなく答えが出ます。

Aがlil_matrixのとき

import numpy
from scipy.sparse import lil_matrix
from scipy.sparse.linalg import spsolve
from scipy import sparse

A = lil_matrix((4, 4))
A[0, 0] = 4
A[0, 2] = -3
A[1, 1] = 1
A[2, 2] = 1
A[3, 3] = 1
A = sparse.csr_matrix(A)
v = numpy.array([ 1, 1, 1, 1])
print spsolve(A, v)

Aがlistのとき

import numpy
from scipy.sparse import lil_matrix
from scipy.sparse.linalg import spsolve
from scipy import sparse

A = [[0]*4 for i in xrange(4)]
A[0][0] = 4
A[0][2] = -3
A[1][1] = 1
A[2][2] = 1
A[3][3] = 1
A = sparse.csr_matrix(A)
v = numpy.array([ 1, 1, 1, 1])
print spsolve(A, v)

Answer 2

spsolveの第1引数はCSCマトリクス形式かCSRマトリクス形式を要求しているようですね。
.tocsr()などで変換が必要ということではないでしょうか？

Comments

[manman]

おっしゃるとおり
A = A.tocsr()
と変換すると、メッセージが消えました。
ありがとうございます。