Numpy多维数组中的运算

我有两个多维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

我们可以在这里使用遮罩吗?