0

2つの2Dnumpy配列でこの機能を実行しようとしています。ステップ1：np.argmax(b, axis=1)インデックスを検索します。ステップ2：検索b[indices] > a[indices] ステップ3：2Dブール配列の戻り値。

私はこれを試しました：

np.where((b>a)&np.argmax(b,axis=1).reshape((3,-1)), True, False)

しかし、サイコロはありません。何か案は？

前もって感謝します。

4

1 に答える 1

3

あなたのコメントに基づいて、私の最も良い理解は次のとおりです。

output = (np.max(b,axis=1)[...,None] == b) & (b > a)

Numpyブロードキャストを使用して、「はb」の部分の行の最大値です。

np.max(b,axis=1)[...,None] == b

またはおそらくもっと明確に：

np.max(b,axis=1)[...,np.newaxis] == b
于 2013-03-26T23:00:41.727 に答える