Works by

Ren's blog

@rennnosuke_rk 技術ブログです

2018-06-29から1日間の記事一覧

【python】numpy配列の特定の条件を満たす要素のインデックスを取得する

メモ。 numpy.arrayオブジェクトの要素のうち、指定した条件を満たす要素は下記のように取得できる。 import numpy as np arr = np.array([[1,2,3],[4,5,6]]) # array([[1, 2, 3], # [4, 5, 6]]) # 特定の条件を満たす要素を取得 arr[arr > 3] # array([4, 5…