如何根据参数数据类型定义不同逻辑的函数？避免重复实现

刚好遇到过类似的需求，Python里有两个非常优雅的方案，完美解决你不想重复代码、又不想写一堆if type(...)判断的痛点：

方案一：用functools.singledispatch实现泛型函数 #

这是Python官方推荐的单分派泛型函数实现方式，专门用来处理不同类型输入的分支逻辑，扩展性极强，而且代码结构清晰。

核心思路是：定义一个基础函数，然后为不同的输入类型注册对应的处理函数，列表类型的处理可以直接复用字符串类型的逻辑，完全避免重复代码。

示例代码：

from functools import singledispatch
# 这里假设你用nltk做词性标注，替换成你自己的标注工具即可
import nltk

@singledispatch
def process_text(input_data):
    # 默认处理逻辑：抛出错误，提示不支持的类型
    raise TypeError(f"Unsupported input type: {type(input_data).__name__}")

@process_text.register(str)
def _process_single_sentence(sentence):
    # 核心逻辑：标注词性，提取形容词、副词、动词
    tokens = nltk.word_tokenize(sentence)
    tagged_tokens = nltk.pos_tag(tokens)
    # 定义目标词性标签（根据你的需求调整）
    target_tags = ('JJ', 'JJR', 'JJS', 'RB', 'RBR', 'RBS', 'VB', 'VBD', 'VBG', 'VBN', 'VBP', 'VBZ')
    return [word for word, tag in tagged_tokens if tag in target_tags]

@process_text.register(list)
def _process_sentence_list(sentences):
    # 直接复用字符串处理逻辑，遍历列表每个元素
    return [process_text(sentence) for sentence in sentences]

使用起来非常简单：

# 处理单个字符串
process_text("The quick brown fox jumps over the lazy dog")
# 处理字符串列表
process_text(["Hello world", "Python is awesome"])

这个方案的优势：

• 代码模块化，新增其他输入类型（比如tuple）只需新增一个@process_text.register装饰的函数即可
• 完全分离不同类型的处理逻辑，可读性强
• 符合Python的官方设计规范

方案二：统一输入处理逻辑，封装核心函数 #

如果不想导入额外模块，也可以用更直观的方式：把核心逻辑封装成一个私有函数，然后在入口函数里做类型判断，统一调度核心逻辑。

示例代码：

import nltk

def _core_process(sentence):
    # 核心逻辑只写一次，所有类型的输入都复用这个逻辑
    tokens = nltk.word_tokenize(sentence)
    tagged_tokens = nltk.pos_tag(tokens)
    target_tags = ('JJ', 'JJR', 'JJS', 'RB', 'RBR', 'RBS', 'VB', 'VBD', 'VBG', 'VBN', 'VBP', 'VBZ')
    return [word for word, tag in tagged_tokens if tag in target_tags]

def process_text(input_data):
    if isinstance(input_data, str):
        return _core_process(input_data)
    elif isinstance(input_data, list):
        return [_core_process(sent) for sent in input_data]
    else:
        raise TypeError(f"Unsupported input type: {type(input_data).__name__}")

这里注意用isinstance而不是type()判断类型，因为isinstance支持继承关系（比如如果有自定义的字符串子类，也能正常处理），比直接判断type更健壮。

方案选择建议 #

• 如果你的系统以后可能需要支持更多输入类型（比如tuple、numpy.ndarray等），优先选方案一，扩展性更好
• 如果只是简单处理str和list两种类型，方案二更轻量，代码更直观

内容的提问来源于stack exchange，提问作者snapper