Partial | 超有用的偏函数

403 阅读3分钟

小知识,大挑战!本文正在参与“程序员必备小知识”创作活动。

什么是偏函数 Partial

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

例子

需求

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

初始代码

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

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

代码运行结果

中国人的代码

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

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

代码运行结果

西安人的代码

当我们这个方法只用来统计中国人的信息的时候,你会发现,getInfocountry 参数的值就固定为“中国”,这个时候我们就需要使用偏函数将 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='小黄')

整体效果

怎么样,你学废了吗?

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