比较numpy矩阵的列和数组 [英] Compare columns of numpy matrix with array

问题描述

我有一个numpy矩阵，想将每列与给定数组进行比较，例如:

I have a numpy matrix and want to compare every columns to a given array, like:

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组成.第二句话是错误，正确，错误.第三个是，否，错.

Now I want to compare every columns of M with v, i.e. I want a matrix with the first column consisting of True, True, True. A second saying False, True, False. A third True, False, False.

我该如何实现?谢谢！

推荐答案

使用广播比较:

>>> M == v[:, None]
array([[ True, False,  True],
       [ True,  True, False],
       [ True, False, False]])

这篇关于比较numpy矩阵的列和数组的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！