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

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

非0要素のインデックス

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

書式

ソースコード

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

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

# 配列の宣言
x = np.array([2, 1, 0])

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

おすすめ記事

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