NumPy で行列を作ってみよう

N 次元配列を作成する

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]]

N 次元配列の型・サイズを確認する

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

サンプルコード

# 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)

参考・出典:The N-dimensional array (ndarray) — NumPy v1.10 Manual