python 二维数组B中所有满足数组A相同位置大于0的数都置为1
时间: 2024-11-27 08:10:57 浏览: 10
python分治法求二维数组局部峰值方法
在Python中,你可以使用列表推导式或者遍历的方式来实现这个操作。假设我们有两个二维数组A和B,你需要找到A中每个元素大于0的位置,并将B中对应位置的值置为1。这里是一个简单的例子:
```python
# 假设 A 和 B 都是二维整数数组
A = [[1, 0, 2], [3, -1, 4]]
B = [[5, 6, 7], [8, 9, 10]]
# 使用列表推导式更新 B
for i in range(len(A)):
for j in range(len(A[i])):
if A[i][j] > 0:
B[i][j] = 1
# 或者直接遍历并修改
for i in range(len(A)):
for j in range(len(A[i])):
if A[i][j] > 0:
B[i][j] = 1
# 输出更新后的 B
print(B)
```
在这个示例中,如果A的某个元素大于0,那么B中对应位置的值就会被设置为1。完成上述操作后,`B` 就会变成只有非零元素的地方才会是1。
阅读全文