python 二维矩阵筛选小于某数的值保留,其余值设置为0
时间: 2024-03-17 21:44:11 浏览: 72
可以使用嵌套的列表推导式来实现这个功能。具体的代码如下:
```python
matrix = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
threshold = 5
result = [[num if num < threshold else 0 for num in row] for row in matrix]
print(result)
```
这段代码中,我们首先定义了一个二维矩阵 `matrix` 和一个阈值 `threshold`。然后,我们使用嵌套的列表推导式来遍历矩阵中的每个元素,并根据其大小来决定是否保留该元素。如果元素的值小于阈值,则保留该值;否则,将其设置为0。最后,我们将处理后的二维列表打印出来。
阅读全文