微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

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语句嵌套在列表推导中!谁知道。

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。