如何解决获取 numpy 数组中所有值的相等掩码
假设我创建了以下 numpy 数组：
a = np.array([[3,3],[1,3,1,3]])
b = np.array([0,2,3])
如果我做 a == 0 我得到：
array([[False,True,False],[False,False,False]])
如果我做 a == 1 我得到：
array([[False,[ True,False]])
等等。但是，如果我想获得一个包含与所有条件 a == n 相关的所有掩码的数组，其中 n 属于 b，我应该如何进行？
np.array([a == n for n in b]) 做了我想要的但看起来不太numpythonic。我也试过 a == b，它只返回 False。
解决方法
只需a == b[:,None,None]，其余的由广播处理：
>>> a == b[:,None]
array([[[False,True,False],[False,False,False]],[[False,[ True,[[ True,True],True]]])
版权声明：本文内容由互联网用户自发贡献，该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请发送邮件至 dio@foxmail.com 举报，一经查实，本站将立刻删除。
