如何在python中增强内置函数的特性?

2024-10-04 15:29:14 发布

您现在位置:Python中文网/ 问答频道 /正文

我是python编程新手, 我想知道如何增强内置函数(Monkeypatch)的特性

例如

我知道sum()内建函数只允许在数字项上使用

>>> sum([4,5,6,7]) #22

我想使sum函数允许项目列表为字符串,如下所示

例如

^{pr2}$

提前谢谢


Tags: 项目函数字符串列表编程数字特性内置
3条回答

不是monkey-patching,只是重新定义了sum,使其同样适用于字符串。在

>>> import __builtin__
def sum(seq, start = 0):
    if all(isinstance(x,str) for x in seq):
        return "".join(seq)
    else:
        return __builtin__.sum(seq, start)
...     
>>> sum([4,5,6,7])
22
>>> sum(['s','t','a','c','k'])
'stack'

你不能像类、对象、模块等那样真正地“monkeypatch”一个函数

这些其他事情最终都归结为一个属性集合,因此用不同的属性替换一个属性,或者添加一个新属性,既简单又有用。另一方面,函数基本上是原子的东西。*

当然,您可以通过替换sum函数来monkeypatch内置模块。但我不认为这是你要的。(如果是,请参见下文。)

无论如何,您不能修补sum,但是您可以编写一个新函数,如果需要,可以使用相同的名称(可能在原始函数的周围加上一个包装器,您会注意到,这正是decorator所做的)。在


但实际上没有办法使用sum(['s','t','a','c','k'])来做你想做的事情,因为sum在默认情况下从0开始并向其添加内容。不能向0添加字符串。**

当然,您总是可以传递显式的start,而不是使用默认值,但是您必须更改调用代码以发送适当的start。在某些情况下(例如,您发送的是一个文本列表显示),这是非常明显的;在其他情况下(例如,在一个通用函数中),它可能不是。这在这里仍然不起作用,因为sum(['s','t','a','c','k'], '')只会引发一个TypeError(尝试并阅读错误以了解原因),但它在其他情况下也可以工作。在

但是没有办法避免必须知道一个合适的起始值,因为sum就是这样工作的。在

如果你仔细想想,sum在概念上等同于:

def sum(iterable, start=0):
    reduce(operator.add, iterable, start)

这里唯一真正的问题是start,对吗?reduce允许您不使用start值,它将从iterable中的第一个值开始:

^{pr2}$

这是sum做不到的。但是,如果你真的想,你可以重新定义sum,这样它就可以了:

>>> def sum(iterable):
...     return reduce(operator.add, iterable)

…或:

>>> sentinel = object()
>>> def sum(iterable, start=sentinel):
...     if start is sentinel:
...         return reduce(operator.add, iterable)
...     else:
...         return reduce(operator.add, iterable, start)

但是请注意,这个sum在整数上比原始的慢得多,它将引发一个TypeError,而不是在空序列上返回{},依此类推。在


如果您真的想要monkeypatch内置函数(而不是仅仅用一个新名称定义一个新函数,或者在模块的globals()中定义一个新函数来隐藏内置函数),下面是一个适用于python3.1+的示例,只要您的模块使用的是普通的globals字典(除非您在嵌入式解释器或exec调用或类似程序中运行,否则它们将是这样):

import builtins
builtins.sum = _new_sum

换言之,与修补任何其他模块一样。在

在2.x中,该模块称为__builtin__。在2.3左右和3.0中,关于如何通过全局变量访问的规则发生了变化。有关详细信息,请参见^{}/^{}。在


*当然这不是真的。函数在其代码对象的顶部有一个名称、闭包单元格列表、文档字符串等。甚至代码对象也是一个字节码序列,您可以使用bytecodehacks或硬编码的黑客。除了sum实际上是一个内置函数,而不是一个函数,因此它甚至没有从Python访问的代码……总之,对于大多数目的来说,它已经足够接近了,可以说函数是原子的东西。在

**当然,可以将字符串转换为某些知道如何将自身添加到整数的子类(通过忽略它们),但实际上,您不想这样做。在

sum已经可以处理任何定义__add__函数的东西。第二个参数是起始点,默认值为0,但您可以将其替换为任何求和结果的“无”版本。例如,从空列表开始添加列表列表:

sum([[1, 2, 3], [4, 5, 6]], [])

退货:

^{pr2}$

所以正常情况下,这实际上是可行的:

sum(['s','t','a','c','k'], '')

但这引发了一个异常,它特别告诉您对字符串使用join。可能是因为它性能更好。在

相关问题 更多 >

    热门问题