对集合执行某些操作，然后返回与集合相同的类型

Question

我想将函数f应用于集合xs，但保留其类型。如果我使用map，我会得到一个“map object”：

def apply1(xs, f):
  return map(f, xs)

如果我知道xs类似于list或tuple，我可以强制它具有相同的类型：

def apply2(xs, f):
  return type(xs)(map(f, xs))

但是，对于namedtuple（我目前正在使用它）来说，这种情况很快就会崩溃——因为据我所知namedtuple需要用解包语法或调用它的_make函数来构造。而且，namedtuple是const，所以我不能遍历所有条目，而只是更改它们。你知道吗

使用dict会产生更多的问题。你知道吗

有没有一种通用的方法来表达这样一个apply函数，它适用于所有可iterable的东西？你知道吗

Answer 1

看起来是^{} decorator的完美任务：

from functools import singledispatch


@singledispatch
def apply(xs, f):
    return map(f, xs)


@apply.register(list)
def apply_to_list(xs, f):
    return type(xs)(map(f, xs))


@apply.register(tuple)
def apply_to_tuple(xs, f):
    try:
        # handle `namedtuple` case
        constructor = xs._make
    except AttributeError:
        constructor = type(xs)
    return constructor(map(f, xs))

之后apply函数可以简单地使用如下

>>> apply([1, 2], lambda x: x + 1)
[2, 3]
>>> from collections import namedtuple
>>> Point = namedtuple('Point', ['x', 'y'])
>>> p = Point(10, 5)
>>> apply(p, lambda x: x ** 2)
Point(x=100, y=25)

虽然我不知道dict对象需要什么样的行为，但是这种方法的伟大之处在于它很容易扩展。你知道吗