获取numpy数组的索引

Question

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

我需要提取 51 以便它可以用来显示它所标记的类。

Answer 1

您似乎正在尝试使用 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 了解更多选项。