KDB/Q技术问题:筛选矩阵中所有元素均大于5的行
如何提取矩阵中所有元素均大于5的行
嘿,这问题挺常见的,我来给你捋捋怎么解决~咱们先明确核心需求:遍历矩阵的每一行,检查这一行里的每一个元素是不是都大于5,只有完全满足这个条件的行,才把它保留下来。
拿你给出的示例矩阵 M = [[1,9,10],[120,50,199],[15,20,7]] 来说,第一行里有个元素1,不满足大于5的要求,所以要排除;第二行和第三行的所有元素都符合条件,这两行就是最终要的结果。
用Python实现的简洁方案
最省心的方式就是用列表推导式,一行代码就能搞定:
M = [[1,9,10],[120,50,199],[15,20,7]] result = [row for row in M if all(num > 5 for num in row)] print(result)
运行这段代码会输出:[[120, 50, 199], [15, 20, 7]]
拆解一下逻辑
row for row in M:逐个遍历矩阵里的每一行all(num > 5 for num in row):用all()函数检查当前行的所有元素是否都满足num > 5的条件——all()会在所有元素都符合要求时返回True,只要有一个不满足就返回False- 只有当
all()返回True时,当前行才会被加入到结果列表中
要是你用其他编程语言,思路也是一致的:遍历每一行,逐元素验证,全满足就保留。比如用R语言可以这么写:
M <- matrix(c(1,9,10,120,50,199,15,20,7), nrow=3, byrow=TRUE) result <- M[apply(M, 1, function(x) all(x > 5)), ] print(result)
内容的提问来源于stack exchange,提问作者kb1964
相关产品推荐
相关产品推荐

