Question

I would like to define a function with the parameter specified as "array of strings" or "array of integers", without looping through the individual array elements to check the type of each. Specifically, I would like the function to only allow passing of the two above-mentioned array types in order to run, in the same way one would be able to specify function(param: str) to only accept strings, for example.

我将单独的职能定义为<条码>功能(第1段:名单/代码>或<条码>功能(第24段:名单/代码)。但是,我无法说明如何同时执行。

Answer 1

You can use typing to hint at what this should be, but to really assert the values, you ll need to check them all
conveniently, all() will short-circuit on the first non-Truthy value

Typing

from typing import Union

def function(param: Union[list[str], list[int]):
    ...

1. 执业

def function(param):
    for test_type in (int, str):
        try:
            if all(type(v) == test_type for v in param):
                break  # escape the for loop
        except TypeError as ex:  # param is not iterable
            if "is not iterable" in str(ex):
                raise TypeError(f"param must be iterable, but got {type(param)})" from ex
            raise ex  # pass through other errors
    else:  # did not break out of for loop
        raise TypeError("function only accepts homogenous iterables of int or str")
    ...  # rest of function

注:这只是允许<代码>第>>为任何iterable的略为允许的,但你可以走到前面,坚持你喜欢的任何东西——但认识到,这一设计将消耗发电机,因此,要收集和维护其价值的工作将会更多(从理论上讲,带有评论和用法的警告器往往比试图强迫他们使用你打算使用的工具)好。

Answer 2

我仍是一个起点,但我将努力:

IDK 如何做到这一点,但你可以检查第一个要素,并与清单的其他部分进行比较。

def func(param: list):
  if type(param[0])!=int and type(param[0])!=str:
    raise Exception("this function accepts only array of integers or array of strings")
  for i in range(1,len(param)):
    if type(param[i]) != type(param[0]):
      raise Exception("this function accepts only array of integers or array of strings")

友情链接