Partial简化函数很好用

Python研究所

发布于 2022-06-17 08:40:01

1970

发布于 2022-06-17 08:40:01

文章被收录于专栏：大飞的部落阁

什么是偏函数 Partial

和装饰器类似，偏函数是用来扩展函数功能的，和装饰器不一样的是，偏函数解决的问题是：当函数中的某些参数值在某些场景下是定值的时候，将其简化。什么意思呢？就是省略某些不变的参数，使得代码会更加简洁。

例子

需求

假设，我们需要写一个方法 getInfo 去获取人类的信息，大概的我们需要国家，城市，姓名，年龄四个参数。

初始代码

# 有中国人和巴基斯坦人
print('\n==================== 以下是中国&巴基斯坦人 ====================')

getInfo('中国','西安','飞',18)
getInfo('巴基斯坦','伊斯兰堡','斯巴达',19)

中国人的代码

# 全是中国人
print('\n==================== 以下都是中国的人 ====================')

getChinaInfo = partial(getInfo,country='中国')
getChinaInfo(city='西安',name='飞',age=18)
getChinaInfo(city='成都',name='瑶',age=19)

西安人的代码

当我们这个方法只用来统计中国人的信息的时候，你会发现，getInfo 的 country 参数的值就固定为“中国”，这个时候我们就需要使用偏函数将 country 参数固定（省略），同时创建一个新的方法 getXianInfo。

# 全是西安人
print('\n==================== 以下都是西安的人 ====================')

# 全部是西安人时，我们不需要传入国家和城市了
# 可以使用偏函数partial来固定参数，同时创建新的函数

getXianIifo = partial(getInfo,country='中国',city='西安')

getXianIifo(name='小明',age=19)
getXianIifo(name='小熊',age=20)

西安18岁人的代码

当我们想要用这个方法来统计西安人信息的时候，我们同样的用偏函数将国家和城市都固定并创建 getXian18Info 方法。

# 全是西安18岁的人
print('\n================== 以下都是西安18岁的人 ==================')
getXian18Info = partial(getInfo,country='中国',city='西安',age=18)

getXian18Info(name='小丽')
getXian18Info(name='小黄')

附全部代码

from functools import partial

def getInfo(country,city,name,age):
    print(f'我叫{name},我是{country}人,我家乡是{city},我今年{age}岁了！')

# 有中国人和巴基斯坦人
print('\n==================== 以下是中国&巴基斯坦人 ====================')

getInfo('中国','西安','飞',18)
getInfo('巴基斯坦','伊斯兰堡','斯巴达',19)


# 全是中国人
print('\n==================== 以下都是中国的人 ====================')

getChinaInfo = partial(getInfo,country='中国')
getChinaInfo(city='西安',name='飞',age=18)
getChinaInfo(city='成都',name='瑶',age=19)

# 全是西安人
print('\n==================== 以下都是西安的人 ====================')

# 全部是西安人时，我们不需要传入国家和城市了
# 可以使用偏函数partial来固定参数，同时创建新的函数

getXianIifo = partial(getInfo,country='中国',city='西安')

getXianIifo(name='小明',age=19)
getXianIifo(name='小熊',age=20)

# 全是西安18岁的人
print('\n================== 以下都是西安18岁的人 ==================')

getXian18Info = partial(getInfo,country='中国',city='西安',age=18)
getXian18Info(name='小丽')
getXian18Info(name='小黄')