使用slicing
,可以这样实现 -
a_out = (a[:-1,:-1] + a[1:,:-1] + a[:-1,1:] + a[1:,1:]).astype(bool)
由于a
已经是一个布尔数组,因此我们可以跳过bool
转换。
在MATLAB上的示例运行 -
>> a = logical([
1, 1, 0, 1, 1, 0
0, 1, 0, 0, 0, 0
1, 1, 0, 1, 1, 1
0, 0, 0, 0, 1, 0
0, 0, 1, 0, 1, 1
0, 0, 0, 1, 1, 0]);
>> a(1:end-1,1:end-1) + a(2:end,1:end-1) + a(1:end-1,2:end) + a(2:end,2:end)
ans =
3 2 1 2 1
3 2 1 2 2
2 1 1 3 3
0 1 1 2 3
0 1 2 3 3
>> logical(a(1:end-1,1:end-1) + a(2:end,1:end-1) + ...
a(1:end-1,2:end) + a(2:end,2:end))
ans =
1 1 1 1 1
1 1 1 1 1
1 1 1 1 1
0 1 1 1 1
0 1 1 1 1
在 NumPy 上运行的示例 -
In [160]: a
Out[160]:
array([[ True, True, False, True, True, False],
[False, True, False, False, False, False],
[ True, True, False, True, True, True],
[False, False, False, False, True, False],
[False, False, True, False, True, True],
[False, False, False, True, True, False]], dtype=bool)
In [161]: (a[:-1,:-1] + a[1:,:-1] + a[:-1,1:] + a[1:,1:])
Out[161]:
array([[ True, True, True, True, True],
[ True, True, True, True, True],
[ True, True, True, True, True],
[False, True, True, True, True],
[False, True, True, True, True]], dtype=bool)
[:-2]
将删除两个元素。 - Stephen Rauch