python - 将一个矩阵中的值替换为另一个矩阵中相同位置的 NA

Question

我有两个形状相同的矩阵，都填充了 float ，但其中一个矩阵在某些位置有 NA。我想让另一个矩阵在相同位置也有 NA。

我能想到一些简单的解决方案:

1)任何类型的条件循环 - 非常慢

2) 将 NA 转换为 0，将矩阵按元素相乘，以便没有 NA 的矩阵在所需位置得到 0，然后按元素除以恢复为原始值，最后将 0 转换为 NA - 速度更快，但仍然低效且笨重

我想找到一种有效的方法来解决这个问题，因为我需要执行这个操作数百万次。

Answer 1

如果A是具有NaN值的矩阵，而B是另一个矩阵。

import numpy as np

# Construct example matrices
A = np.random.random((3,3))
B = np.random.random((3,3))
# Set two values to np.nan
A[0,:2] = np.nan
# Set values in B to np.nan in the same locations that they occur in A
B[np.isnan(A)] = np.nan