【NumPy】任意の値をもつ要素数をカウント

この記事では、Python言語とNumPyを用いて、任意の値をもつ要素数をカウントする方法をソースコード付きで解説します。

任意の値をもつ要素数

PythonモジュールNumPyでは、whereメソッドで、条件を満たす要素番号を抽出できます。
今回はそれを用いて任意の値をもつ要素数を求めてみました。

サンプルプログラムのソースコードです。

# -*- coding: utf-8
import numpy as np


# 配列生成
A = np.array([0, 1, 1, 1, 2, 0, 0])

# 値が2の要素数
num =  len(np.where(A==1)[0])

# 結果表示
print(num) # 3