相关疑难解决方法(0)

测试2d numpy数组的成员资格

我有两个相同大小的2D阵列

a = array([[1,2],[3,4],[5,6]])
b = array([[1,2],[3,4],[7,8]])
Run Code Online (Sandbox Code Playgroud)

我想知道a中的b行.

所以输出应该是:

array([ True,  True, False], dtype=bool)
Run Code Online (Sandbox Code Playgroud)

不做:

array([any(i == a) for i in b])
Run Code Online (Sandbox Code Playgroud)

因为a和b很大.

有一个函数可以执行此操作但仅适用于1D数组:in1d

python numpy

10
推荐指数
1
解决办法
2624
查看次数

标签 统计

numpy ×1

python ×1