python数组处理之最值与下标问题
作者:北海鲤
这篇文章主要介绍了python数组处理之最值与下标问题,具有很好的参考价值,希望对大家有所帮助。如有错误或未考虑完全的地方,望不吝赐教
python最值与下标
最大值的下标
winner = np.argmax(scores)
多个最大值的下标
(np.argwhere返回数组中非0元素的索引)
winners = np.argwhere(scores == np.amax(scores)) winners = winners.flatten()
数组排序并返回原下标
from operator import itemgetter Lst = np.array([2,3,1,4,5]) indices, L_sorted = zip(*sorted(enumerate(L), key=itemgetter(1)))
元素逐个求平方
err=[num*num for num in Lst]
创建空list矩阵
Lst= [[] for _ in range(K)]
Lst= lsts = [[[] for _ in range(b)] for _ in range(a)]
列表中不重复元素的个数
a = ['lunch', 'afternoon', 'lunch', 'dinner', 'dinner', 'dawn', 'lunch'] print(len(set(a))) # 4
axis=0代表行向,=1代表列向
a = [[1,2],[3,4]] print(np.sum(a,axis=0)) [4 6] print(np.sum(a,axis=1)) [3 7]
求最大值及其下标
编写程序,找出给定的n个数中的最大值及其对应的最小下标(下标从零开始)。
程序代码:
ls=input().split() lst=list(map(int,ls)) #利用map函数把ls中的元素换成整型,再用列表输出 t=max(lst) a=lst.index(t) print(t,a)
总结
以上为个人经验,希望能给大家一个参考,也希望大家多多支持脚本之家。