python - 从矩阵创建掩码

Question

你好，我有一个像这样的矩阵

A=[[nan, 4, nan],[3 , 7 , 8],[nan, 23, nan]]

我想从矩阵A中获取一个掩码，如下

mask=[[nan, 0, nan],[0, 0, 0],[nan, 0, nan]]

为此我已经尝试过:

import numpy as np
A=[[nan, 4, nan],[3 , 7 , 8],[nan, 23, nan]]
mask=A
mask[np.isfinite(A)]=0

但这也会删除矩阵 A 的数值。

Answer 1

您需要复制 A 才能保留 A 中的值，请参阅:https://docs.python.org/2/library/copy.html

在你的情况下，这将是

A=[[nan, 4, nan],[3 , 7 , 8],[nan, 23, nan]]
mask=np.array(A.copy())
mask[~np.isnan(A)] = 0