Matplotlib scatterplot 그래픽에서 x y 좌표를 얻는 방법

Question

분산 형 그래픽을 호스트하는 GUI에 Widget이 있습니다. 나는 pick event을 포함 할 수 있었지만, 단지 print('yahoo') 만 플롯을 클릭하여 x 좌표를 얻을 수는 없습니다. 내 코드는 다음 기사를 기반으로합니다 : see link (섹션 : 간단한 피킹 예제). 산점도에서 점을 클릭하여 점의 좌표 (예 : x = 20, y = 50, 내 5 점)를 얻는 방법은 무엇입니까?

프런트 엔드 코드 :

from PyQt5 import QtCore, QtGui, QtWidgets 

class Ui_MainWindow(object): 
    def setupUi(self, MainWindow): 
     MainWindow.setObjectName("MainWindow") 
     MainWindow.resize(595, 393) 
     self.centralwidget = QtWidgets.QWidget(MainWindow) 
     self.centralwidget.setObjectName("centralwidget") 
     self.gridLayout = QtWidgets.QGridLayout(self.centralwidget) 
     self.gridLayout.setObjectName("gridLayout") 
     self.pushButton = QtWidgets.QPushButton(self.centralwidget) 
     self.pushButton.setMaximumSize(QtCore.QSize(100, 16777215)) 
     self.pushButton.setObjectName("pushButton") 
     self.gridLayout.addWidget(self.pushButton, 0, 0, 1, 1) 
     self.widget = QtWidgets.QWidget(self.centralwidget) 
     self.widget.setMinimumSize(QtCore.QSize(0, 200)) 
     self.widget.setStyleSheet("background-color: rgb(255, 255, 255);") 
     self.widget.setObjectName("widget") 
     self.gridLayout.addWidget(self.widget, 0, 1, 1, 1) 
     MainWindow.setCentralWidget(self.centralwidget) 
     self.menubar = QtWidgets.QMenuBar(MainWindow) 
     self.menubar.setGeometry(QtCore.QRect(0, 0, 595, 21)) 
     self.menubar.setObjectName("menubar") 
     MainWindow.setMenuBar(self.menubar) 
     self.statusbar = QtWidgets.QStatusBar(MainWindow) 
     self.statusbar.setObjectName("statusbar") 
     MainWindow.setStatusBar(self.statusbar) 

     self.retranslateUi(MainWindow) 
     QtCore.QMetaObject.connectSlotsByName(MainWindow) 

    def retranslateUi(self, MainWindow): 
     _translate = QtCore.QCoreApplication.translate 
     MainWindow.setWindowTitle(_translate("MainWindow", "MainWindow")) 
     self.pushButton.setText(_translate("MainWindow", "PushButton")) 


if __name__ == "__main__": 
    import sys 
    app = QtWidgets.QApplication(sys.argv) 
    MainWindow = QtWidgets.QMainWindow() 
    ui = Ui_MainWindow() 
    ui.setupUi(MainWindow) 
    MainWindow.show() 
    sys.exit(app.exec_()) 

백엔드 코드 :

import sys 
import matplotlib.pyplot as plt 
from PyQt5 import QtCore, QtGui, QtWidgets 
from PyQt5.QtWidgets import QMainWindow, QApplication, QWidget, QSizePolicy 
from matplotlib.backends.backend_qt5agg import FigureCanvasQTAgg as FigureCanvas 
from matplotlib.backends.backend_qt5agg import NavigationToolbar2QT as NavigationToolbar 
from matplotlib.figure import Figure 
from front_end import Ui_MainWindow 


class Graph_init(FigureCanvas): 

    def __init__(self, parent=None): 

     fig = Figure() 
     fig.patch.set_facecolor("None") 
     self.axes = fig.add_subplot(111) 
     self.compute_initial_figure() 

     FigureCanvas.__init__(self, fig) 
     self.setParent(parent) 

     FigureCanvas.setSizePolicy(self, QSizePolicy.Expanding, QSizePolicy.Expanding) 
     FigureCanvas.updateGeometry(self) 

     FigureCanvas.mpl_connect(self, 'pick_event', self.onclick) 

    def onclick(self, event): 
     print('yahoo') 
     thisline = self.event.artist 
     xdata = thisline.get_xdata() 
     ydata = thisline.get_ydata() 


class Graph_populate(Graph_init): 

    def compute_initial_figure(self): 

     x = [10,20,30,40,50] 
     y = [100,50,150,200,75] 
     size = [1000,2000,3000,5000,2000] 

     self.axes.scatter(x,y,s=size,color='blue', picker=1) 
     self.axes.patch.set_facecolor('None') 


class GUI(QMainWindow, Ui_MainWindow): 
    def __init__(self, parent=None): 
     super(GUI, self).__init__(parent) 
     self.setupUi(self) 
     self.sc = Graph_populate(self.widget) 
     self.gridLayout.addWidget(self.sc, 0, 1, 1, 1) 


if __name__ == '__main__': 
    app = QApplication(sys.argv) 
    prog = GUI() 
    prog.showMaximized() 
    sys.exit(app.exec_()) 

당신은 산점도 음모를 꾸미고있다

Answer 1

감사합니다. 따라서 산란 점이 처리되는 pick_event_demo의 세 번째 부분을 살펴볼 수 있습니다.

아이디어는 선택된 데이터 포인트의 인덱스를 가져 와서 해당 인덱스에서 데이터를 가져 와서 좌표를 얻는 것입니다. 그 데이터 포인트의 coordintes는, (내가 질문을 이해하는 방법 임) 마우스 클릭의 그 아니라는 것을

는

def onclick(self, event): 
    ind = event.ind[0] 
    data = event.artist.get_offsets() 
    xdata, ydata = data[ind,:] 
    print ((xdata, ydata)) 

참고.