【NumPy】非0要素のインデックス

この記事では、Python言語とNumPyを用いて、配列の非0要素のインデックス(要素番号・位置)を取り出す方法をソースコード付きで解説します。

非0要素のインデックス

NumPyでは、配列の非0要素のインデックス(要素番号・位置)を取り出す機能が用意されています。

書式

説明
index = numpy.nonzero(x) 配列xの値が0以外の要素のインデックスを取得します。

ソースコード

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

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

def main():

    # 配列の宣言
    x = np.array([2, 1, 0]) 
    
    # 非0要素のインデックス
    print(np.nonzero(x)) # (array([0, 1]),)


if __name__ == '__main__':
    main()

おすすめ記事

Python入門 サンプル集
NumPy入門 サンプル集

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