在Python中,可以使用列表推导式来智能地删除零行以获得一个方阵。方阵是一个行数和列数相等的二维数组。
以下是一个示例代码:
# 原始矩阵
matrix = [
[1, 2, 3],
[0, 0, 0],
[4, 5, 6]
]
# 删除零行后的方阵
square_matrix = [row for row in matrix if any(row)]
print(square_matrix)
输出结果为:
[[1, 2, 3], [4, 5, 6]]
这段代码使用列表推导式遍历原始矩阵的每一行,通过any(row)
判断该行是否存在非零元素。如果存在非零元素,则将该行保留在新的方阵中。最终得到的square_matrix
即为删除零行后的方阵。
推荐的腾讯云相关产品:腾讯云函数(云函数是事件驱动的无服务器计算服务,可帮助您在腾讯云上构建和运行应用程序,无需管理服务器),产品介绍链接地址:https://cloud.tencent.com/product/scf
领取专属 10元无门槛券
手把手带您无忧上云