记录[i for (i, b) in enumerate(x) if b ]的用法

x = [-1,0,1,2,3]
test1 = [i for i in enumerate(x)]
test2 = [i for (i, b) in enumerate(x)]
test3 = [i for (i, b) in enumerate(x) if b]
print(test1)
print(test2)
print(test3)

运行结果如下:

[(0, -1), (1, 0), (2, 1), (3, 2), (4, 3)]
[0, 1, 2, 3, 4]
[0, 2, 3, 4]

enumerate(x)函数遍历x中所有项,生成默认从0开始的下标,如-1的下标为0,0的下标为1,…,存放方式为:(下标,内容),如(3,2)。
test1:生成同时记录下标和内容的数组。
test2:只生成x下标的数组。
test3:通过if b语句判断,内容是否为True,如果为True,则记录其下标,生成记录x中为True内容下标的数组。当b = 0,if b 为False。

你可能感兴趣的:(python)