scipy.ndimage.

查找对象#

scipy.ndimage.find_objects(input, max_label=0)[源代码][源代码]#

在标记的数组中查找对象。

参数:
输入int 的 ndarray

包含由不同标签定义的对象的数组。值为0的标签将被忽略。

max_labelint, 可选

input 中搜索的最大标签。如果未给出 max_label,则返回所有对象的位置。

返回:
对象切片元组列表

一个元组列表,每个元组包含 N 个切片(N 是输入数组的维度)。切片对应于包含对象的最小平行六面体。如果缺少一个数字,则返回 None 而不是切片。标签 l 对应于返回列表中的索引 l-1

注释

此功能对于隔离3-D数组中无法“透视”的感兴趣区域非常有用。

示例

>>> from scipy import ndimage
>>> import numpy as np
>>> a = np.zeros((6,6), dtype=int)
>>> a[2:4, 2:4] = 1
>>> a[4, 4] = 1
>>> a[:2, :3] = 2
>>> a[0, 5] = 3
>>> a
array([[2, 2, 2, 0, 0, 3],
       [2, 2, 2, 0, 0, 0],
       [0, 0, 1, 1, 0, 0],
       [0, 0, 1, 1, 0, 0],
       [0, 0, 0, 0, 1, 0],
       [0, 0, 0, 0, 0, 0]])
>>> ndimage.find_objects(a)
[(slice(2, 5, None), slice(2, 5, None)),
 (slice(0, 2, None), slice(0, 3, None)),
 (slice(0, 1, None), slice(5, 6, None))]
>>> ndimage.find_objects(a, max_label=2)
[(slice(2, 5, None), slice(2, 5, None)), (slice(0, 2, None), slice(0, 3, None))]
>>> ndimage.find_objects(a == 1, max_label=2)
[(slice(2, 5, None), slice(2, 5, None)), None]
>>> loc = ndimage.find_objects(a)[0]
>>> a[loc]
array([[1, 1, 0],
       [1, 1, 0],
       [0, 0, 1]])