python中有没有办法遍历列表推导中的矩阵

如何解决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语句嵌套在列表推导中！谁知道。