代码之家 › 专栏 › 技术社区 › Nico Schlömer David Maze

查找包含任何列表的NumPy数组行

numpy arrays python

Nico Schlömer David Maze · 技术社区 · 8 年前

a 和list/set/1D NumPy数组 b .我想找到这些行 一 其中包含任何 b

import numpy as np

a = np.array([
    [1, 2, 3],
    [4, 5, 3],
    [0, 1, 0]
    ])

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

# result: [True, False, True]

有什么提示吗?

1 回复 | 直到 8 年前

Divakar 8 年前

你可以使用 np.in1d 从中查找任何元素的匹配项 b 在中的每个元素中 a 现在 np.in1d型 将使数组变平,因此我们需要在以后重新整形。最后,因为我们想找到 ANY 沿中的每一行进行匹配 一 使用 np.any 沿着每一行。因此,我们将有这样一个实现-

np.in1d(a,b).reshape(a.shape).any(axis=1)

推荐文章

serlingpa · 如何准备我的数据以避免无法推断频率

1 年前

Guillaume · 使用操作从Python列表创建numpy数组

2 年前

user19657580 · 在Python中打印两个numpy数组的列表

2 年前

user19657580 · Python中数组中具有相同元素的索引求和

2 年前

mikanim · 改进二维余弦函数的numpy功能

2 年前

Klimt865 · 在Python中将数组列表转换为列表列表

2 年前

theduker · 计算平均绝对误差时,If语句中赋值前引用的局部变量

2 年前

Lynn · 如果列包含Python中的特定字符串,则从列中删除值

2 年前

JasonX · 运行减法计算

2 年前

Jan Hrubec · 选择numpy数组的前n个元素

2 年前