我是一个新用户，所以我需要一些指导。假设我有这个3x3矩阵：
[[ -1, 1, 1], [ -1, 1, 1], [ -1, -1, -1]]
[[ -1, 1, 1],
[ -1, 1, 1],
[ -1, -1, -1]]
例如，我希望检测与以下内容的匹配：
[[any, 1, 1], [any, 1, 1], [any, any, any]]
[[any, 1, 1],
[any, 1, 1],
[any, any, any]]
(在本例中，匹配将为True)Numpy有没有一个简单的函数/诀窍可以做到这一点？
True
r7knjye21#
您可以使用遮罩和NaN：
a = np.array([[ -1, 1, 1], [ -1, 1, 1], [ -1, -1, -1]])m = np.array([[np.nan, 1, 1], [np.nan, 1, 1], [np.nan, np.nan, -1]])out = ((a == m) | np.isnan(m)).all()# True
a = np.array([[ -1, 1, 1],
[ -1, -1, -1]])
m = np.array([[np.nan, 1, 1],
[np.nan, 1, 1],
[np.nan, np.nan, -1]])
out = ((a == m) | np.isnan(m)).all()
# True
1条答案
按热度按时间r7knjye21#
您可以使用遮罩和NaN：