scipy.ndimage.label连通算法

最新推荐文章于 2024-10-17 11:39:06 发布

文子阳

最新推荐文章于 2024-10-17 11:39:06 发布

阅读量715

点赞数 1

分类专栏：数据分析三剑客文章标签：算法 scipy python

本文链接：https://blog.csdn.net/wenxingchen/article/details/128924310

版权

数据分析三剑客专栏收录该内容

7 篇文章 0 订阅

订阅专栏

这段代码演示了如何利用scipy的ndimage库对numpy数组进行标注(label)和查找连通组件(find_objects)，以确定连续1值的起始和结束位置。它特别适用于处理二进制数据或图像分割问题。

摘要由CSDN通过智能技术生成

目标：获取连续的1值的位置

import numpy as np
from scipy import ndimage

x = np.asarray([0, 0, 0, 0, 1, 1, 1, 1, 1, 0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1])
# 标记连接部分
labels, num_features = ndimage.label(x)
# 获取连通部分的首尾索引
objects = ndimage.find_objects(labels)
for i in objects:
    print("首", i[0].start)
    print("步长", i[0].step)
    print("尾", i[0].stop)