NumPy で行列を作ってみよう

NumPy では、ndarray クラスを用いて、多次元行列を作成することができます。以下コードでは、2×3 の 2 次元の行列を作成します。以下コード中の np.int32 は、配列の要素が 32 bit の整数であることを示しています。

サンプルコード

import numpy as np

x = np.array([[1, 2, 3], [4, 5, 6]], np.int32)

print x

実行結果

>>> import numpy as np

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

>>> print x

[[1 2 3]

[4 5 6]]

作成した配列は以下のようにして、型やサイズ（大きさ）を確認できます。

サンプルコード

# 2 x 3 行列を作成

x = np.array([[1, 2, 3], [4, 5, 6]], np.int32)

# 配列自体の型を確認

print type(x)

# 配列の要素の型を確認

print x.dtype

# 配列のサイズを確認

print x.shape

実行結果

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

>>> print type(x)

>>> print x.dtype

int32

>>> print x.shape

(2L, 3L)