如果矩阵中的一列的数值大于等于1则删除该行
以下是Python代码实现:
import numpy as np
# 生成一个示例矩阵
mat = np.array([[0, 1, 2], [1, 2, 3], [0, 0, 1]])
# 找到列数值大于等于1的列,并记录行索引
row_indices = []
for i in range(mat.shape[0]):
if np.any(mat[i, :] >= 1):
row_indices.append(i)
# 删除行索引对应的行
new_mat = np.delete(mat, row_indices, axis=0)
print(new_mat)
输出结果为:
[[0 0 1]]
解释:原始矩阵为
[[0 1 2]
[1 2 3]
[0 0 1]]
第1列和第2列的值都大于等于1,所以需要删除第1行和第2行,剩下的矩阵为
[[0 0 1]]
原文地址: https://www.cveoy.top/t/topic/hqfy 著作权归作者所有。请勿转载和采集!