这两个Python函数可以被一个接受列表或元组参数的泛型函数替换吗?

您可以使用 isinstance 函数检查输入的类型,然后返回相应的类型。

def negative_sequence(a):
    if isinstance(a, list):
        return [-e for e in a]
    elif isinstance(a, tuple):
        return tuple(-e for e in a)
    else:
        raise TypeError("Input must be a list or a tuple")

# Example usage:
print(negative_sequence([1, 2, 3]))  # Output: [-1, -2, -3]
print(negative_sequence((1, 2, 3)))  # Output: (-1, -2, -3)

您可以在运行时确定类型

def negative(a):
    return type(a)((-e for e in a))

所以我们也可以使用 Iterable 类使其更健壮

from collections.abc import Iterable
import numpy as np


def negative(a):
    if isinstance(a, Iterable):
        result = (-e for e in a)
        if isinstance(a, list):
            return list(result)
        elif isinstance(a, tuple):
            return tuple(result)
        elif isinstance(a, np.ndarray):
            return np.negative(a)
        else:
            raise TypeError("Not supported iterable type.")
    else:
        raise TypeError("Interables only.")