0
我Python中的新手, 我需要找到一個2維列表的第二列的最大元素的索引如下
3 | 23
7 | 12
5 | 42
1 | 25
所以輸出應該是42的索引即2
我試圖將列表轉換爲numpy數組,然後使用argmax,但徒勞
謝謝!找到一個2維列表的第二列的最大元素的索引
我Python中的新手, 我需要找到一個2維列表的第二列的最大元素的索引如下
3 | 23
7 | 12
5 | 42
1 | 25
所以輸出應該是42的索引即2
我試圖將列表轉換爲numpy數組,然後使用argmax,但徒勞
謝謝!找到一個2維列表的第二列的最大元素的索引
>>> import numpy as np
>>> a = np.asarray([[3, 7, 5, 1], [23, 12, 42, 25]])
>>> a
array([[ 3, 7, 5, 1],
[23, 12, 42, 25]])
>>> a[1]
array([23, 12, 42, 25])
>>> np.argmax(a[1])
2