python中对list列表中的元素或者list类型的matrix进行操作

 今天做了下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]]

猜你喜欢

转载自blog.csdn.net/qq_26593695/article/details/93850895