配列 ベクトル 行列 統計

NumPyで行列の全要素数を求める(size)

最終更新日 2022.11.27

NumPy では行列の全要素数を size で求めます。

import numpy as np

A = np.array([[1, 2],
              [3, 4]])

B = np.array([[0, 0, 1],
              [1, 0, 0]])

C = np.array([[1, 1, 1],
              [1, 1, 1],
              [1, 1, 1]])

a = A.size
b = B.size
c = C.size

print(a)  # 4
print(b)  # 6
print(c)  # 9

行列の行数と列数を表す shape と比べてください。

import numpy as np

A = np.array([[0, 0, 1],
              [1, 0, 0]])

shape = A.shape
size = A.size

print(shape)  # (2, 3)
print(size)  # 6

size は shape の要素をかけた数に等しくなります。