gpt4 book ai didi

python - 比较两个包含 NaN 的矩阵并屏蔽两个矩阵中至少其中一个在 Python 中包含 NaN 的元素值

转载 作者:行者123 更新时间:2023-12-01 02:03:59 25 4
gpt4 key购买 nike

我有两个 3D 矩阵(A 和 B),两者都包含随机元素中的 NaN。我正在比较这两个矩阵,在其中至少一个包含 NaN 的地方,我希望它们都包含 NaN。换句话说,如果它们在该索引处尚未包含 NaN,我想用 NaN 替换该索引值。有没有一种有效的方法可以使用 python 函数来做到这一点?

import numpy as np

# Create the fake variables A and B. Here is what A and B look like initially:
A = np.array([[1, 2, np.nan], [4, np.nan, 6], [np.nan, 8, 9]])
B = np.array([[1, 2, 3], [4, 5, np.nan], [np.nan, 8, 9]])

# What I want A and B to look like in the end:
A
array([[ 1., 2., nan],
[ 4., nan, nan],
[ nan, 8., 9.]])

B
array([[ 1., 2., nan],
[ 4., nan, nan],
[ nan, 8., 9.]])

最佳答案

您需要numpy.isnan()boolean indexing .

>>> A[np.isnan(B)] = np.nan
>>> B[np.isnan(A)] = np.nan

关于python - 比较两个包含 NaN 的矩阵并屏蔽两个矩阵中至少其中一个在 Python 中包含 NaN 的元素值,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/49257094/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com