我有两个多维numpy数组。如果第一个数组中的对应元素为零,我想将第二个数组中的条目转换为NaN。下面是手动模拟相同内容的示例:(可以通过编程方式完成)
import numpy as np
a = np.random.rand(4,5)
a[0][0] = 0
a[1][0] = 0
a[1][1] = 0
b = np.random.rand(4,5)
b[0][0] = np.nan
b[1][0] = np.nan
b[1][1] = np.nan
我们可以在这里使用遮罩吗?