Python:找到ndarray数组中最大元素的位置

import numpy as np
from sklearn import datasets
from sklearn.linear_model import LogisticRegression
from sklearn.model_selection import train_test_split
# a = np.array([5,6,7,8,1,2,3,3,4,2,1])
# b = np.unique(a)
# print(b)
# X,y = datasets.load_iris(return_X_y=True)
# X_train,X_test, y_train, y_test = train_test_split(X,y,test_size=0.2,random_state=0)
# model = LogisticRegression(solver='newton-cg',penalty='l2')
# model.fit(X_train,y_train)
# print(model.predict_proba(X_test))

a = np.array([1,2,3,5,2,1,4])
print(np.argmax(a))

别指望用一次就会了，呵呵！

个把月你就忘了~真的~