如何将用户在Matplotlib中输入的点传递给np.array？

问题描述：

我想创建一个程序，允许用户首先在Matplotlib图中输入点，然后使用这些点创建一个Voronoi图。如何将用户在Matplotlib中输入的点传递给np.array？

找出2个部分，但没有连接。我如何传递用户在Voronoi部分输入并使用它的点？（我只需要知道如何改变points = np.random.rand(20,2)到用户输入的点。）

中绘制点：

import matplotlib.pyplot as plt 

fig = plt.figure() 
ax = fig.add_subplot(111) 
ax.set_xlim([0, 10]) 
ax.set_ylim([0, 10]) 

def onclick(event): 
    print('x=%d, y=%d, xdata=%f, ydata=%f' % 
      (event.x, event.y, event.xdata, event.ydata)) 
    plt.plot(event.xdata, event.ydata, 'bo') 
    fig.canvas.draw() 

cid = fig.canvas.mpl_connect('button_press_event', onclick) 
plt.show()`

创建Voronoi图：

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

fig = plt.figure() 

points = np.random.rand(20,2) 

vor = Voronoi(points) 

voronoi_plot_2d(vor) 

plt.show()

你想产生与Voronoi图或情节新的情节，在同样的情节？ – ImportanceOfBeingErnest

我现在并不在意这一点。如果可能在同一个情节中，它会很棒，但现在并不重要。 –

答

您可以使用一个类来存储点，然后在同一个图中绘制图。在下面的例子中，你会用鼠标左键放置你的点，并绘制voronoi图，点击鼠标右键。如果您稍后想为图表添加新点，则可以这样做。

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

fig, ax = plt.subplots() 
ax.set_xlim([0, 10]) 
ax.set_ylim([0, 10]) 

class V(): 
    def __init__(self, ax = None): 
     self.ax = ax 
     if not self.ax: self.ax = plt.gca() 
     tx = "left click to place points\nright click to plot voronoi diagram" 
     self.text = self.ax.text(0.1,0.9, tx, transform=self.ax.transAxes, 
           ha="left", va="top") 
     self.cid = fig.canvas.mpl_connect('button_press_event', self.onclick) 
     self.points = [] 

    def onclick(self, event): 
     self.text.set_visible(False) 
     if event.button == 1: 
      print('x=%d, y=%d, xdata=%f, ydata=%f' % 
        (event.x, event.y, event.xdata, event.ydata)) 
      plt.plot(event.xdata, event.ydata, 'bo') 
      self.points.append((event.xdata, event.ydata)) 
     else: 
      self.voronoi() 
     fig.canvas.draw() 

    def voronoi(self): 
     self.ax.clear() 
     vor = Voronoi(self.points) 
     voronoi_plot_2d(vor, ax=self.ax) 


v = V(ax=ax) 
plt.show()

非常感谢！你从字面上挽救了我的一天。 –

答

你可以只保存您的xy对列表中，并在稍后将其转换为数组：

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


fig = plt.figure() 
ax = fig.add_subplot(111) 
ax.set_xlim([0, 10]) 
ax.set_ylim([0, 10]) 


points = [] 
def onclick(event): 
    print('x=%d, y=%d, xdata=%f, ydata=%f' % 
      (event.x, event.y, event.xdata, event.ydata)) 
    points.append([event.xdata,event.ydata]) 
    plt.plot(event.xdata, event.ydata, 'bo') 
    fig.canvas.draw() 

cid = fig.canvas.mpl_connect('button_press_event', onclick) 
plt.show() 

fig2 = plt.figure() 

vor = Voronoi(np.array(points)) 

voronoi_plot_2d(vor) 

plt.show()

As lon g当用户点击图表时，点数将被添加到列表中，直到用户关闭原始数字。之后，创建一个新图形并绘制Voronoi图。希望这可以帮助。

如何将用户在Matplotlib中输入的点传递给np.array？

相关推荐