代码之家 › 专栏 › 技术社区 › iagerogiannis

在3D-np数组中搜索1D-np数组

numpy python-3.x python

iagerogiannis · 技术社区 · 3 年前

在下面的示例中,我尝试在3D np数组中搜索1D np数组。

import numpy as np


a = np.array([[[1, 2], [3, 4]],
              [[5, 6], [3, 2]],
              [[2, 1], [7, 8]]])

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

print(np.all(a == b, axis=1))
# print(np.where(np.all(a == b, axis=(1,1))))

[[False False]
 [False False]
 [False False]]

这意味着我得到了布尔表的正确维数,但由于某些原因,结果不是预期的结果。

1 回复 | 直到 3 年前

akuiper 3 年前

你需要 axis=2 ,即第三轴:

np.all(a == b, axis=2)

[[ True False]
 [False False]
 [False False]]

或者您可以指定 axis=-1 对于最后一个轴:

np.all(a == b, axis=-1)

[[ True False]
 [False False]
 [False False]]

推荐文章

Aaron Green · 我的python程序无法识别数据库的存在,即使它在那里

1 年前

danial · 如何在多个字符串的每个位置找到最频繁的字符

2 年前

Henry · 使用Python将json重新格式化为键值对

2 年前

eymentakak · json字典类型错误:字符串索引必须是整数

2 年前

Qubix · 从熊猫数据帧创建相对熵矩阵

2 年前

FÄÅ ÛÅ · 字典、列表和字符串

2 年前

OrbitDuster · 如何使用gmail api在python中打印gmail正文?

2 年前

guiguilecodeur · 如何删除我的词汇表中的重复元素

2 年前

Susheel P M · 这是关于if-else语句[关闭]

2 年前

Slartibartfast · 关于Python版本安装

2 年前