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

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

任意の値をもつ要素数

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

ソースコード

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

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

    
def main():
    # 配列生成
    A = np.array([0, 1, 1, 1, 2, 0, 0])
    
    # 値が2の要素数
    num =  len(np.where(A==1)[0])
    
    # 結果表示
    print(num) # 3
   
    
if __name__ == "__main__":
    main()

【おすすめ記事】
Python入門 サンプル集
NumPy入門 サンプル集

シェア&フォローお願いします!