如何键入提示返回 zip object 的问题？

Question

我有一个 function 需要任意数量的列表（或任何可迭代的，就此而言）并将它们排序为一个。 代码如下所示：

def sort_as_one(*args):
    return zip(*sorted(zip(*args)))
def main():
    list1 = [3, 1, 2, 4]
    list2 = ["a", "b", "d", "e"]
    list3 = [True, False, True, False]
    result = sort_as_one(list1, list2, list3)
    # <zip object at ...>
    print(result)
    list1, list2, list3 = result
    print(list1, list2, list3)
if __name__ == "__main__":
    main()

如何准确键入提示 function output？

Answer 1

zip object 是一个迭代器——它遵循迭代器协议。 习惯上，您可能只是这样输入提示。 在这种情况下，您希望使用类型变量将其作为泛型类型提示：

import typing

T = typing.TypeVar("T")

def sort_as_one(*args: T) -> typing.Iterator[T]:
    return zip(*sorted(zip(*args)))

请注意，如果您使用可变参数 arguments，您必须只接受一个类型。 在这种情况下，您可以做的最好的事情是使用Any而不是T 。 但是，如果您希望能够将 function 与 static 类型检查器一起使用，则应考虑仅在代码中使用上述 function。