程序员

Learn Python 3:高阶函数

2017-10-19  本文已影响60人  SheHuan

把函数作为参数传入,这样的函数称为高阶函数

以下是常用的高阶函数:

1、map

一般情况map()函数接收两个参数,一个函数(该函数接收一个参数),一个序列,将传入的函数依次作用到序列的每个元素,并返回一个新的Iterator(迭代器)

例如有这样一个list:['pYthon', 'jaVa', 'kOtlin'],现在要把list中每个元素首字母改为大写,其它的改为小写,可以这样操作:

>>> def f(s):
...    return s.title()
... 
>>> l = map(f, ['pYthon', 'jaVa', 'kOtlin'])
>>> list(l)
['Python', 'Java', 'Kotlin']

由于l是一个Iterator,所以通过list()函数计算出整个序列,并返回一个list,当然也可以通过next(l)逐个取出l中的元素。

上边的f(),函数只有一个参数,如果有多个参数,那么map()函数就可以接收一个函数以及对应函数参数个数的序列,例如map()函数接收的函数有两个参数时,可以再传入两个序列,这时两个序列就是函数的两个参数值,例如:

>>> def f(x, y):
...    return x + y
... 
>>> list(map(f, [1, 2, 3],[4, 5, 6]))
[5, 7, 9]

2、reduce

和map()用法类似,reduce把传入的函数作用在一个序列上,但传入的函数需要接收两个参数,传入函数的计算结果继续和序列的下一个元素做累积计算。

例如有一个list,里边的元素都是字符串,要把它拼接成一个字符串:

>>> from functools import reduce
>>> def f(x, y):
...    return x + y
... 
>>> reduce(f, ['ab', 'c', 'de', 'f'])
'abcdef'

3、filter

filter()同样接收一个函数和一个序列,然后把传入的函数依次作用于序列的每个元素,如果传入的函数返回true则保留元素,否则丢弃,最终返回一个Iterator

例如一个list中元素有纯字母、纯数字、字母数字组合的,我们要保留纯字母的:

>>> def f(s):
...    return s.isalpha()
... 
>>> l = filter(f, ['abc', 'xyz', '123kg', '666'])
>>> list(l)
['abc', 'xyz']

4、sorted

sorted()函数就是用来排序的,同时可以自己定义排序的规则。

例如:

>>> sorted([6, -2, 4, -1])
[-2, -1, 4, 6]

这里默认按照数字从小到大的顺序排列。如果需要修改排序规则为按照绝对值排列可以新增一个key参数,代表排序规则的函数:

>>> sorted([6, -2, 4, -1], key=abs)
[-1, -2, 4, 6]

因为默认是升序排列,如果要改为降序可使用reverse参数:

>>> sorted([6, -2, 4, -1], key=abs, reverse=True)
[6, 4, -2, -1]

如果是字符串序列则默认按照元素首字母的ASCII的大小,升序排列:

>>> sorted(['Windows', 'iOS', 'Android'])
['Android', 'Windows', 'iOS']

如果需要忽略大小写,可以指定key=str.lower或者key=str.upper来完成。

有一个姓名、身高对应关系的list,其中每个元素是tuple,现在要按照身高升序排列list:

>>> d = [('Tom', 170), ('Jim', 175), ('Andy', 168), ('Bob', 185)]
>>> def by_height(t):
...     return t[1]
... 
>>> sorted(d, key=by_height)
 [('Andy', 168), ('Tom', 170), ('Jim', 175),  ('Bob', 185)]
上一篇 下一篇

猜你喜欢

热点阅读