np.argmax (a, axis = None, out = None) function: return the index of the maximum value of axis dimension
a: input an array of array type
axis: compare the entire array when the parameter is None, the parameter is 0 and the comparison is by column 1 Compare by row
map () will map the specified sequence according to the provided function.
The first parameter function calls the function function with each element in the parameter sequence, returning a new list containing the value returned by each function function.
