for 循环生成列表

1.简单的for...[if]...语句

list1 = [1,2,3,4,5,6,7,8,9]
new_list = [x for x in list1 if x % 2 == 0]
print(new_list)

输出：
[2, 4, 6, 8]

2.把双层列表生成单层新列表

list1 = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
new_list = [x for temp_list in list1 for x in temp_list]
print(new_list)

输出
[1, 2, 3, 4, 5, 6, 7, 8, 9]

生成批量

batches = [frames[i:i + args.batch_size] for i in range(0, len(frames), args.batch_size)]

0
32
64
96
128

frames是一个列表，里面存放着181张HxWx3的图片，然后frames[i:i + args.batch_size]，i依次取0，32，64，96，128，args.batch_size=32，所以

batches=

frames[0:32],

frames[32,64]

frames[64,96]

frames[96,128]

frames[128,160]
剩下最后21个单独成一个批量

 

一些好的参考博客

Python for循环生成列表_白水baishui的博客-CSDN博客_python循环结果保存为列表