python - 列表过滤索引

Question

我有以下数字列表:

List = [1, 2, 3, 4, 5, 6, 15]

我想要那些是 n 的倍数的数字的索引，所以我这样做:

def indexes(List, n):
      # to enumerate the numbers
      E = enumerate(List)
      # filtering tuples
      F = list(filter(lambda x:  x[1] % n == 0, E))
      return [ i[0] for i in F]

     indexes(List, 2)
     [1, 3, 5]

没关系，但是当我添加变量 m 时会发生什么？

def Index( L, n, m):
      # enumeration
      E = enumerate(L)
      # filtering tuples
      F_n = list(filter(lambda x: x[1]%n == 0, E))
      F_m = list(filter(lambda x: x[1]%m == 0, E))
      L_n = [ l[0] for l in F_n]
      L_m = [ J[0] for J in F_m]
      return L_n + L_m

  >>>Index(List, 2, 5):
        [1, 3, 5]

为什么该代码不返回 [1, 3, 5, 4, 6]？

错在哪里？

以及如何创建返回该列表的函数？

Answer 1

您可以将列表理解 与enumerate 方法结合使用。也可以申请extended iterable unpacking运算符以便传递所需数量的参数。

List = [1, 2, 3, 4, 5, 6, 15]
def indexes(List, *vars):
   return [index for index, item in enumerate(List) for i in vars if item % i == 0 ]

print(indexes(List, 2, 5))

输出

[1, 3, 5, 4, 6]