gpt4 book ai didi

python - 基本 NumPy 数据比较

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

我有一个以 2D 数组排列的 N 维值数组。像这样的东西:

import numpy as np
data = np.array([[[1,2],[3,4]],[[5,6],[1,2]]])

我还有一个值x,我想将其与每个数据点进行比较,并且我想获得一个 bool 值的二维数组,显示我的数据是否等于x.

x = np.array([1,2])

如果我这样做:

data == x

我明白了

# array([[[ True,  True],
# [False, False]],
#
# [[False, False],
# [ True, True]]], dtype=bool)

我可以轻松地将这些组合起来以获得我想要的结果。但是,我不想迭代每个切片,尤其是当 data.shape[2] 较大时。我正在寻找的是一种直接获得的方法:

array([[ True,  False],
[False, True]])

对于这个看似简单的任务有什么想法吗?

最佳答案

好吧,(data == x).all(axis=-1) 给了你你想要的。它仍在构造一个 3 维结果数组并对其进行迭代,但至少该迭代不是 Python 级别的,因此它应该相当快。

关于python - 基本 NumPy 数据比较,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/10418325/

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