python - 以适用于str和unicode的方式使用translate的python方法是什么？

Question

我有一个函数可以清除 str 中的特定字符:

def _clean_name(name):
    return name.translate(None, "\n ").replace('.', '_')

现在这个函数也必须支持unicode，所以我最终得到了这个版本:

def _clean_name(name):
    if type(name) is unicode:
        return name.translate({ord('\n'): None, ord(' '): None, ord('.'): ord('_')})
    else:
        return name.translate(None, "\n ").replace('.', '_')

但这对我来说并不太Pythonic。有人有更好的解决方案吗？

Answer 1

这可能更像是一种hackish的做法，而不是 Pythonic 的做法 - 但一个优点是它将消除代码重复，并且可能应该适用于 unicode 的子类(您绝对需要检查如果需要的话!)。

因此，利用类型对象是可调用的，并且 unicode 和 str 都实现 join 操作:

def _clean_name(name):
    return type(name)().join(
#          ^^^^^^^^^^^^
#           build an empty str or unicode object
        [{'.': '_', '\n':'', ' ':''}.get(c,c) for c in name]
    )