2017-02-20 12 views
0

私は形の数が少ない配列を持っています(1, 7, 3) 3列の中で最も高い要素を持つ行を探したいと思います。例:232は3列目で最大ですので、出力する必要があります[196 228 232]どうすればいいですか?私はnp.argmax試みたが、例の配列です。ここnumpy配列の最大桁を見つけよう

を失敗しました:

[[[218 204 204] 
[344 194 31] 
[284 140 108] 
[196 228 232] 
[324 196 28] 
[224 228 57] 
[174 250 144]]] 
+0

多分このリンクは、[最大のhttp://stackoverflow.com/questions/12403238/maximum-values-along-axis-of-numpy-ndarray – phunsukwangdu

+0

可能な複製を助けるかもしれませんNumpy ndarrayの軸に沿った値?](http://stackoverflow.com/questions/12403238/maximum-values-along-axis-of-numpy-ndarray) – Evert

+0

a.max(軸= 2)は '[218 344 284 232 324 228 250] '各行の最大値です。問題のように最大の行[2]要素を持つ要素が必要です – Wideem

答えて

1

argmaxはここに正しい考えです。それを一歩一歩しましょう。

row_nr = np.argmax(data[0, :, 2]) 

これは、3番目の列を選択し、最大値のインデックスを検出します。 それは、この行を選択したまま:

data[0, row_nr, :] 
+0

私の質問を読んで答えてくれてありがとう、ありがとう。それは今完璧な意味がある – Wideem

関連する問題