Python3 求最大/小值及索引值 Numpy
阿新 • • 發佈:2019-01-28
若尋找一個列表的最大(小)值及其對應的索引:
list = [9, 12, 88, 14, 25]
max_list = max(list) # 最大值的索引
max_index = list.index(max(list)) # 返回最大值
# 最小的話 max換成min
若是numpy中arrary的型別
# 把array轉換為list,按以上方法 import numpy as np
a= np.array([9, 12, 88, 14, 25]) list_a = a.tolist() list_a_max_list = max(list_a) #返回最大值 max_index = list_a.index(max(list_a)) # 返回最大值的索引
min/max是python內建的函式
np.argmin/np.argmax是numpy庫中的成員函式
(可適合處理numpy.ndarray物件,可選的引數是axis=0或者1)
import numpy as np a = np.array([1, 2, 3, 4]) b = np.array((5, 6, 7, 8)) c = np.array([[11, 2, 8, 4], [4, 52, 6, 17], [2, 8, 9, 100]]) print(a) print(b) print(c) print(np.argmin(c)) print(np.argmin(c, axis=0)) # 按每列求出最小值的索引 print(np.argmin(c, axis=1)) # 按每行求出最小值的索引 # 最小的話 min換成max