如何解决python中有没有办法遍历列表推导中的矩阵
我有以下矩阵:
from random import randint
n,m = 5,7
matrix = [[randint(-9,9) for j in range(m)] for i in range(n)]
我如何在列表理解的帮助下遍历此矩阵而没有自己的循环?
我的目标是获取小于k
且位于奇数行的元素列表,总的来说,我想汇总以下代码以列出理解:
sample = []
for i in range(n):
for j in range(m):
if matrix[i][j] < k and i % 2 != 0:
sample.append(matrix[i][j])
我试图做到,但是没有用:
sample = [elem for elem in row for row in matrix if elem < k]
解决方法
我相信这与您的示例具有相同的作用,但是在列表理解方面:
import itertools
sample = [ matrix[i][j] for (i,j) in itertools.product(range(n),range(m)) if matrix[i][j] < k and i % 2 != 0 ]
,
我刚刚在做一些Python研究,碰巧看到一些代码使我知道可以在列表推导中使用嵌套的for循环。有了这些知识,并且刚刚回答了这个问题,我想我应该把它归功于这个世界,以便写下第二个答案。
我认为这是一个优势,因为A)它不需要使用必须导入的模块,而B)它更简单,更酷:
sample = [ matrix[i][j] for i in range(n) for j in range(m) if matrix[i][j] < k and i % 2 != 0 ]
因此if
语句嵌套在列表推导中!谁知道。
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。