python求取一个数组的(最大值/最小值)及其(索引)

x = [3, 2.2, 7.4, 6, 4]
min_index, min_number = min(enumerate(x), key=operator.itemgetter(1))
# 输出 1   2.2

max_index, max_number = max(enumerate(x), key=operator.itemgetter(1))
# 输出 2   7.4

解析:

enumerate()函数用于将一个可遍历的数据对象(如列表、元组或字符串)组合为一个索引序列,同时列出数据和数据下标。

x = [3, 2.2, 7.4, 6, 4]
list(enumerate(x))
# 输出 [(0, 3), (1, 2.2), (2, 7.4), (3, 6), (4, 4)]

operator.itemgetter()函数用于获取对象的哪些维的数据,参数为想要取的一些维度序号。

x = [3, 2.2, 7.4, 6, 4]
b1 = operator.itemgetter(2, 1)
b1(x) # 输出 (7.4, 2.2)

b2 = operator.itemgetter(3)
b2(x) # 输出 6

max()函数有一个应用很巧妙的参数key,在这里定义为operator.itemgetter(1),表示对enumerate(x)每个元素的第一维做比较(从0维开始),然后返回第一维值最大的元素,即包含索引数值

key参数还有其他的巧妙应用:

# 获取数组中绝对值最大的数
x = [3, 2.2, -7.4, 6, 4]
max(x, key=abs) # 返回 -7.4


# lambda表达式形式
x = [3, 2.2, -7.4, 6, 4]
max(x, key=lambda x: abs(x)) # 返回 -7.4

Reference

http://www.runoob.com/python/python-func-enumerate.html

https://www.cnblogs.com/zhoufankui/p/6274172.html

https://www.cnblogs.com/zhangwei22/p/9892422.html

你可能感兴趣的:(Python)