python - 支持单个字符串和数组作为一个参数

Question

这个问题已经有答案了:

Single string or list of strings in a method [duplicate] (6 个回答)

已关闭 5 年前。

我extended a method带有附加字符串参数(平均值):

def classification_report(y_true, y_pred, labels=None, target_names=None,
                          sample_weight=None, digits=2, average='weighted'):

所以正常用法是这样的:

classification_report(y_true, y_pred, average='micro')

来自 Github 的评论，审阅者建议也支持多个值:average=['micro', 'weighted', 'macro']

在 Python 中解决这个问题的合适方法是什么？我知道参数可以有任何类型。但是如何使用字符串列表处理潜在的调用呢？参数的类型是什么？ Union[str, List[str]] ？

目前我只期望一个字符串，这很简单。但如果我也允许字符串列表，我应该首先使用 type(average) 检查类型吗？然后相应地处理这些值？有没有一种优雅的方法将参数变量转换为统一类型，例如 averages=listify(average) ，它接受一个字符串或一个字符串列表，并且总是返回一个字符串列表？

Answer 1

这是 pandas 的示例:

if level is not None:
        if not isinstance(level, (tuple, list)):
            level = [level]

这是来自方法 pandas.DataFrame.reset_index(level=None, drop=False, inplace=False, col_level=0, col_fill='') 的代码，其中 level 预计为int、str、tuple 或 list，默认 None。这个简单的代码只是确保参数现在存储在列表中，无论它是如何给出的。