［Python］行列を作る

NumPyライブラリのndarrayクラスを使う。以下は、2行3列の行列を作る例。

>>> import numpy as np
>>> np.array([[1, 2, 3], [4, 5, 6]])
array([[1, 2, 3],
       [4, 5, 6]])

行列から要素を取り出すには、[行番号,列番号]と指定する。ただし、行番号も列番号も0から始まるので注意。上の例で、2行3列目の要素である6を取り出すには、以下のようにする。

>>> mx = np.array([[1, 2, 3], [4, 5, 6]])
>>> mx[1, 2]
6