获取numpy数组的索引

时间:2019-06-07 12:37:07

标签: python numpy

x=pred_vec.argmax(axis=-1) 
x
o/p-array([51], dtype=int64)

我需要提取51,以便可以用来显示其被标记为的类。

1 个答案:

答案 0 :(得分:0)

似乎您正在尝试使用python和numpy(基于标记??)来查找最大值的索引。如果是这样,请尝试以下操作:

In [1]: import numpy as np                                                                                                                                                                                                                                        

In [3]: x = np.random.randint(1, 200, (50))                                                                                                                                                                                                                       

In [4]: x                                                                                                                                                                                                                                                         
Out[4]: 
array([186, 117, 150,  68, 118, 197, 169,  48,  90,  67, 117,  90, 133,
       141, 175, 194,  21, 107, 103,  61, 154,  87,  42,  59,  67,  66,
       198,  76, 190,   7, 188, 146, 158,  42, 106, 157,   1, 114, 185,
        58, 169,  89, 120,  81,  28,  49, 197,  73, 169, 142])

In [5]: index_max = np.argmax(x)                                                                                                                                                                                                                                  

In [6]: x[index_max]                                                                                                                                                                                                                                              
Out[6]: 198

看看sorting, searching and counting page以获得更多选项。