Voronoi图的边缘：如何从scipy.spatial.Voronoi对象中获取形式为（point1，point2）的边缘？

Question

我已经花费了大量时间尝试从scipy.spatial.Voronoi图中获取边缘，但一直没有成功。这是主要文档： http://docs.scipy.org/doc/scipy-dev/reference/generated/scipy.spatial.Voronoi.html

如果您像这样创建Voronoi图：

points = np.array([[0, 0], [0, 1], [0, 2], [1, 0], [1, 1], [1, 2],
[2, 0], [2, 1], [2, 2]]) //Or feel free to use any set of points

那么您可以访问以下对象属性：

vor.regions
vor.max_bound
vor.ndim
vor.ridge_dict
vor.ridge_points
vor.ridge_vertices
vor.npoints
vor.point_region
vor.points

但是如何将它们结合起来以获取二维沃罗诺伊图中的边缘形式 (point1, point2) 尚不清楚？我知道边缘存在，因为你可以绘制沃罗诺伊图及其边缘和顶点，方法如下：

voronoi_plot_2d(vor)
plt.show()

这张图片清晰地展示了voronoi边缘-如何获取它们以及它们的起始点和终止点？如果我只获取实心边缘（而不是在图表上无限延伸的虚线），那也没问题。

- cgnorthcutt

---

对于快速阅读者的简单答案：vor.ridge_vertices 提供了索引到 vor.vertices 数组以获取边缘的位置。 - cgnorthcutt

Answer 1

看一下ridge_vertices属性：

    ridge_vertices  (list of list of ints, shape (nridges, *))
        Indices of the Voronoi vertices forming each Voronoi ridge.

该列表中的每个元素都是一对整数。每个整数都是vertices列表中的索引。因此，每个元素定义了在Voronoi图中要绘制的线。-1的索引表示一个“无限远”的点。

下面是绘制Voronoi图线的脚本：

import numpy as np
import matplotlib.pyplot as plt
from scipy.spatial import Voronoi


points = np.array([[0, 0], [0, 1], [0, 2],
                   [1, 0], [1, 1], [1, 2],
                   [2, 0], [2, 1], [2, 2]])

vor = Voronoi(points)


fig = plt.figure()

# Mark the Voronoi vertices.
plt.plot(vor.vertices[:,0], vor.vertices[:, 1], 'ko', ms=8)

for vpair in vor.ridge_vertices:
    if vpair[0] >= 0 and vpair[1] >= 0:
        v0 = vor.vertices[vpair[0]]
        v1 = vor.vertices[vpair[1]]
        # Draw a line from v0 to v1.
        plt.plot([v0[0], v1[0]], [v0[1], v1[1]], 'k', linewidth=2)

plt.show()

它创建了以下内容：

- Warren Weckesser

---

太棒了！对于未来的观众，主要的见解是vor.ridge_vertices提供了进入vor.vertices数组的索引，这就是你得到边缘的方法！太棒了。谢谢。 - cgnorthcutt

非常感谢！我刚刚发现 plt.hold(True) 已经被弃用了，而且这个调用是多余的。来源：链接 - Martin

@Martin，感谢你告诉我 hold(True)已经被弃用了。我已将其从代码中删除。 - Warren Weckesser