次のような状況があります。D (8,4) 行列を定義しましょう:

ランダムな列を設定しましょう:

この値を空のリスト Sp に保存します。

この列の行列 D の最大値のインデックスを決定したいと思います。

これまでのところ、すべて正常に動作しています。ここで、同じプロセスを実行したいのですが、列 j に対する行列 D の行のサブセットに対してだけです (簡単にするために、固定したままにします)。私はこれを試しました

問題は、返されるインデックスが、行列 D の要素に対応する人ではなく、行列 D[a,j] のインデックスであることです。結果は、値 58 に対応する z=2 になるはずですが、返された値は z=1 でした。これは、行列 D[a,j] 内の値 58 の位置です。

このプロセスを行うときは、元の行列 (D) に対するインデックスが必要です。何か案が？私はいくつかのアプローチを見てきましたが、これを行う簡単な方法が必要だと思いますか?

行のサブセットが常に [1,2,3] または [5,6,7] であるとは限らないため、この質問と同じではないことに注意してください。私の例では、[2,7,3,4] などの一般的なサブセットを扱いますが、明確な開始 - 終了はありません。

関数 Argmax を使用して、文字列の配列で最大サイズの文字列を取得しています。文字列に単一の文字の繰り返しが含まれていると、結果が奇妙になります。

例えば：

np.array を見てみましょう:

このコードを使用して、値がしきい値より大きい最初の列の最初の要素を見つけます。

しきい値 = 1 を取得すると、期待どおりのインデックスが得られます。

しかし、2 より大きい値を選択すると、出力は 0 になります。どの要素もしきい値を満たさない場合に備えて、最初の値ではなく最後の値を取得したいため、これは私のプログラムを台無しにします。

その場合、配列の最後の値を取得する効率的な方法はありますか?

編集：

ここでこれがどのように役立つか実際にはわかりません: Numpy: How to find first non-zero value in every column of a numpy array?

むしろ、0 ではなく argmax を返すようなものを探していFalseます。

@Divakar と @Achintha Ihalage によるソリューションの比較

したがって、@Divakarのソリューションは超高速です! どうもありがとう！