我有一个 numpy 矩阵,想要将每一列与给定的数组进行比较,例如:
M = np.array([1,2,3,3,2,1,1,3,2]).reshape((3,3)).T
v = np.array([1,2,3])
现在我想将 M 的每一列与 v 进行比较,即我想要一个第一列由 True、True、True 组成的矩阵。第二个说假,真,假。第三个对,错,错。
我该如何实现这一目标? 谢谢!
最佳答案
使用广播比较:
>>> M == v[:, None]
array([[ True, False, True],
[ True, True, False],
[ True, False, False]])
关于python - 比较 numpy 矩阵的列与数组,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/53903206/