时间戳

首页 > 解决方案 > numpy数组中给定值的凸包 - python 2.7

python - numpy数组中给定值的凸包 - python 2.7

问题描述

对于给定的数组（如下面的建议）和给定的值（此处为 0），我想计算在同一个凸包上可以关联多少个 0。

大批 ：

1 2 4 5 8 9 7
4 0 0 7 5 6 8
6 5 0 4 3 5 2
1 0 0 5 7 0 6
2 3 5 7 8 9 4

对于建议的数组，解决方案应该得到：[5, 1]因为可以在第二行和第四行以及第二/第三列之间识别五个 0 的第一个“模式”。存在只有一个 0 的第二种模式。

你知道如何得到它吗？我知道如何计算 0 的数量。也许我可以用 numpy array 的 mask 选项做点什么？

最好的问候和感谢您的帮助，

标签： pythonarraysnumpyconvex-hull

解决方案

您正在寻找的是数组中的“连接组件”。Scipy 有一个方便的label功能：

import numpy as np
from scipy.ndimage.measurements import label

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

labs, n_components = label(a==0)

component_sizes = [np.sum(labs==i) for i in range(1, n_components+1)]

print(component_sizes)

印刷[5, 1]

推荐阅读