今天做了下leetcode上的题,是要对matrix进行操作,而这个matrix给的是list的类型不是numpy的array,所以要对List进行操作。形如[[1,0,0],[1,1,0],[0,0,0]]
问题:
现在我要对里面的元素都乘以5,但是如果我用for循环直接遍历的话会很耗时间,如果是个1000的矩阵就很浪费资源。而直接使用list*5这种操作,只是对list做了一个copy五次的操作,并不能得到想要的结果。并且观察到,这个list里面的元素还是个List.
解决办法:
使用map可以进行迭代和lambda构造列表推导式。
代码如下:
matrix=[[1,0,0],[1,0,1]]
f=lambda x: x*2
b=[list(map(f,i)) for i in matrix]
print(b)
输出结果为:
[[2, 0, 0], [2, 0, 2]]