编写一个可以接受值和任何类型的可迭代类型的函数

I wonder if it is possible to write the following code in a very pythonic way. The code is simplificated. It should accept a single values for each input, as well as a sequence or a numpy array. The problem is that a single value does not have any I wonder if it is possible to write the followi