带有参数列表的python函数

Question

我想定义一个函数，找出满足某些条件的列表的索引。但是总是有一些错误:

TypeError: 'int' object is not iterable

下面是我的代码

def find_index(x,*arr):
    for i in len(arr):
        if abs(i-x) < 1e-5:
            j = arr.index(i)
            return j

如何指定 arr 是列表还是 numpy 数组？ (抱歉，我不知道如何在这个编辑器中格式化代码)

非常感谢。

Answer 1

(第一个)问题是for i in len(arr)。 len(arr) 返回一个不可迭代的整数。您可能很想使用 range(len(arr))，但不要屈服。还有更好的选择!我们可以在迭代时跟踪值的索引。通常，此函数可能类似于:

def find_index(x, arr):
    for i, val in enumerate(arr):
        if abs(val - x) < 1.e-5:
            return i

现在您可以在任何 可迭代对象上调用它。例如:

find_index(5, [1, 2, 3, 5, 60])  # passing a list
find_index(5, (0, 5+1e-6, 10))  # passing a tuple