Question

我有以下功能示例（数字列表的平均值）：

def avg(obs):
    return (1. / len(obs)) * np.sum(obs)

avg([1,2,3,4,5])
3.0

我很有兴趣了解如何使用单个可迭代参数和两个或多个参数，如max()

max(1,2,3,4,5)
5
max([1,2,3,4,5])
5

Answer 1

首先，除非你有充分的理由，否则你不应该这样做; min和max执行此操作是因为它们的二进制形式非常常见，例如sum仅适用于iterables。

那就是说，以下是诀窍。

from collections import Sized

def avg(*obs):
    if len(obs) == 1 and isinstance(obs[0], Sized):
        obs = obs[0]
    return (1. / len(obs)) * sum(obs)

（Sized是一个抽象基类，它概括了列表，集合，dicts，元组，NumPy数组和各种其他容器。它确保定义了len。）

如何构建函数以处理单个可迭代或两个或多个参数

1 个答案: