Partial简化函数很好用

什么是偏函数 Partial

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

例子

需求

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

初始代码

代码语言：javascript复制

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

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

中国人的代码

代码语言：javascript复制

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

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

西安人的代码

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

代码语言：javascript复制

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

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

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

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

西安18岁人的代码

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

代码语言：javascript复制

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

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

附全部代码

代码语言：javascript复制

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='小黄')

怎么样，你学废了吗？

以上就是今天的全部内容了，感谢您的阅读，我们下节再会。

partial 统计 装饰器

0 人点赞