python - 如何从 numpy 矩阵中删除 nan 和 inf 值？

Question

这是我的代码

import numpy as np
cv = [[1,3,4,56,0,345],[2,3,2,56,87,255],[234,45,35,76,12,87]]
cv2 = [[1,6,4,56,0,345],[2,3,4,56,187,255],[234,45,35,0,12,87]]

output = np.true_divide(cv,cv2,where=(cv!=0) | (cv2!=0))
print(output)`

我得到 Nan 和 inf 值。我尝试删除不同的方法，一旦我删除了 Nan，然后我删除了 Inf 值并将它们替换为 0。但是我需要将它们一起替换!有什么方法可以一起替换它们吗？

Answer 1

您可以使用以下掩码替换 NaN 和无限值:

output[~np.isfinite(output)] = 0

>>> output
array([[1.        , 0.5       , 1.        , 1.        , 0.        ,
        1.        ],
       [1.        , 1.        , 0.5       , 1.        , 0.46524064,
        1.        ],
       [1.        , 1.        , 1.        , 0.        , 1.        ,
        1.        ]])