python – 过滤列表的索引

2023年1月26日 178次阅读

我有以下数字列表：

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]？

这是什么错误？

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

最佳答案 您可以将列表推导与枚举方法结合使用.

此外,您可以应用
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]
点赞