当前位置 主页 > 网站技术 > 代码类 >

    Python3 实现减少可调用对象的参数个数

    栏目:代码类 时间:2019-12-20 21:10

    问题

    一个被其他python代码使用的callable对象,可能是一个回调函数或者是一个处理器,由于其参数太多,导致调用时出错。

    解决方案

    如果需要减少某个函数的参数个数,可以使用functools.partial() 。 partial() 函数允许给一个或多个参数设置固定的值,减少接下来被调用时的参数个数。 假设一个函数有很多参数:

    def func(a, b, c, d):
      print(a, b, c, d)

    使用 partial() 函数来固定某些参数值:

    from functools import partial
    
    s1 = partial(func, 1) # a = 1
    print(s1(2, 3, 4))
    (1, 2, 3, 4)
    
    s2 = partial(func, d=100)  # d = 100
    print(s2(1, 2, 3))
    (1, 2, 3, 100)
    
    s3 = partial(func, 1, 2, d=100) # a = 1, b = 2, d = 100
    print(s3(3))
    (1, 2, 3, 100)
    

    可以看出 partial()函数固定某些参数,并返回一个新的callable对象。这个新的callable接受未赋值的参数, 然后跟之前已经赋值过的参数合并起来,最后将所有参数传递给原始函数。

    讨论

    假设有一个点的列表points,表示(x, y)坐标元组。现在需要以点(4, 3)为基点,根据点和基点之间的距离来排序points中所有的点。

    定义一个函数,用来计算两点之间的距离:

    import math
    
    points = [(1, 2), (3, 4), (5, 6), (7, 8)]
    pt = (4, 3)
    
    def distance(p1, p2):
      x1, y1 = p1
      x2, y2 = p2
      return math.hypot(x2 - x1, y2 - y1)
    

    列表的 sort() 排序方法,可以接受一个关键字参数,设置自定义排序的逻辑, 但是它只能接受一个单个参数的函数,很显然,distance()函数不符合条件,通过使用 partial() 函数可以解决这个问题:

    points.sort(key=partial(distance, pt))
    print(points)
    [(3, 4), (1, 2), (5, 6), (7, 8)]

    很多时候 partial() 能实现的效果,其实,lambda表达式也能实现。

    points.sort(key=lambda x: distance(pt, x))
    print(points)
    [(3, 4), (1, 2), (5, 6), (7, 8)]

    以上这篇Python3 实现减少可调用对象的参数个数就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持IIS7站长之家。