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

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

非0要素のインデックス

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

書式

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

ソースコード

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

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

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

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

おすすめ記事

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

コメント