gpt4 book ai didi

python - 比较 numpy 矩阵的列与数组

转载 作者:行者123 更新时间:2023-12-01 01:20:20 26 4
gpt4 key购买 nike

我有一个 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/

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