我有两个多维 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
我们可以在这里使用 mask 吗?
最佳答案
像你说的那样写:
b[a==0] = np.nan
关于numpy - Numpy 多维数组中的操作,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/61863582/