Базовая карта Matplotlib: удаление нанесенных точек

Я пытаюсь создать кнопку, которая отображает случайную точку на базовой карте matplotlib. Единственная проблема заключается в том, что я хочу, чтобы ранее нанесенные точки были удалены (т.е. на карте должна быть только одна точка в данный момент времени).

Я думал, что после инициализации пустой базовой карты я мог бы сохранить ее в переменной (newMap = self.map), поэтому, когда я хочу построить новую точку, я мог бы просто построить ее, используя newMap (т.е. newMap.plot(x, y, 'o')). Насколько я могу судить, изменение сохраняется в исходной базовой карте self.map.

Кто-нибудь знает, как «очистить» базовую карту, чтобы ранее нанесенные точки не сохранялись?

Полный код:

from PySide import QtCore, QtGui

import matplotlib
matplotlib.use('Qt4Agg')
matplotlib.rcParams['backend.qt4']='PySide'
from matplotlib.backends.backend_qt4agg import FigureCanvasQTAgg as FigureCanvas
from matplotlib.backends.backend_qt4agg import NavigationToolbar2QT as NavigationToolbar
from matplotlib.figure import Figure
from mpl_toolkits.basemap import Basemap

from random import randint

class MapWidget(QtGui.QWidget):
    def __init__(self, parent=None):
        super().__init__(parent)
        self.figure = Figure()
        self.canvas = FigureCanvas(self.figure)
        self.layout = QtGui.QVBoxLayout(self)
        self.mplToolbar = NavigationToolbar(self.canvas, self, coordinates=False)
        self.layout.addWidget(self.canvas)
        self.layout.addWidget(self.mplToolbar)
        self.axes = self.figure.add_subplot(111)
        self.setLayout(self.layout)
        # Create our Basemap
        self.map = Basemap(projection='robin', lon_0=0, ax=self.axes, resolution='c')
        self.map.drawcoastlines()
        self.map.drawcountries()
        self.canvas.draw()
        self.show()

    def changeMap(self, lat, lon):
        # I'm saving the supposedly 'clean' Basemap to a variable and plotting on that
        newMap = self.map
        x, y = newMap(lon, lat)
        newMap.plot(x, y, 'o')
        self.canvas.draw()


class Ui_MainWindow(object):
    def plot(self):
        # Plot on some random coordinate
        self.Map.changeMap(randint(-90, 90), randint(-180, 180))

    def myChanges(self):
        # Initialize our Basemap
        self.Map = MapWidget()
        self.layoutMap = QtGui.QVBoxLayout(self.widget)
        self.layoutMap.addWidget(self.Map)
        # Plot a new point when the button is clicked
        self.pushButton.clicked.connect(self.plot)

    # The following is all PySide code, not very relevant to the question
    def setupUi(self, MainWindow):
        MainWindow.setObjectName("MainWindow")
        MainWindow.resize(800, 600)
        self.centralwidget = QtGui.QWidget(MainWindow)
        self.centralwidget.setObjectName("centralwidget")
        self.horizontalLayout_2 = QtGui.QHBoxLayout(self.centralwidget)
        self.horizontalLayout_2.setObjectName("horizontalLayout_2")
        self.horizontalLayout = QtGui.QHBoxLayout()
        self.horizontalLayout.setObjectName("horizontalLayout")
        self.pushButton = QtGui.QPushButton(self.centralwidget)
        self.pushButton.setMaximumSize(QtCore.QSize(100, 16777215))
        self.pushButton.setObjectName("pushButton")
        self.horizontalLayout.addWidget(self.pushButton)
        self.widget = QtGui.QWidget(self.centralwidget)
        self.widget.setObjectName("widget")
        self.horizontalLayout.addWidget(self.widget)
        self.horizontalLayout_2.addLayout(self.horizontalLayout)
        MainWindow.setCentralWidget(self.centralwidget)
        self.menubar = QtGui.QMenuBar(MainWindow)
        self.menubar.setGeometry(QtCore.QRect(0, 0, 800, 21))
        self.menubar.setObjectName("menubar")
        MainWindow.setMenuBar(self.menubar)
        self.statusbar = QtGui.QStatusBar(MainWindow)
        self.statusbar.setObjectName("statusbar")
        MainWindow.setStatusBar(self.statusbar)

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

        self.myChanges()

    def retranslateUi(self, MainWindow):
        MainWindow.setWindowTitle(QtGui.QApplication.translate("MainWindow", "MainWindow", None, QtGui.QApplication.UnicodeUTF8))
        self.pushButton.setText(QtGui.QApplication.translate("MainWindow", "Plot", None, QtGui.QApplication.UnicodeUTF8))


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

python matplotlib matplotlib-basemap pyside matplotlib-widget
person Arda Arslan    schedule 30.06.2017    source источник

Ответы (1)


arrow_upward
2
arrow_downward

Удалить старую точку и нарисовать новую — это то же самое, что просто изменить положение одной точки.

Таким образом, вы должны создать пустой график и сохранить его в переменной класса self.point, = self.map.plot([],[], 'o').

class MapWidget(QtGui.QWidget):
    def __init__(self, parent=None):
        super(MapWidget, self).__init__(parent)
        # ....
        self.point, = self.map.plot([],[], 'o')
        self.canvas.draw()
        self.show()

Затем вы используете свою карту (обратите внимание, что newmap = self.map не является копией, это ссылка на ту же карту, поэтому нет необходимости создавать здесь новые переменные) и устанавливаете новые данные для графика, хранящегося в self.point.

def changeMap(self, lat, lon):
    x, y = self.map(lon, lat)
    self.point.set_data(x, y)
    self.canvas.draw_idle()
person ImportanceOfBeingErnest    schedule 30.06.2017