兰州大学2019级数据科学编程期末复习（二）

列表推导

二维列表生成

columns,rows  =  5 , 5
a=[[y for y in range(0,columns)] for x in range(0,rows)]
#[[0, 0, 0, 0, 0], [0, 0, 0, 0, 0], [0, 0, 0, 0, 0], [0, 0, 0, 0, 0], [0, 0, 0, 0, 0], [0, 0, 0, 0, 0]]

找出在x中而不在y的元素

x,y=[i for i in range(0,11)],[i for i in range(5,16)]
result=[i for i in x if not (i in y )]
#[0,1,2,3,4]

列表推导式

[ element for var in collection if condition]