Plotting 2D scatter points in a PyQt5 UI

[DRLing2021]

Hello,

I notice that vedo does fantastic job on 3D plotting. Now we are making a UI based on PyQt5 to show both 3D and 2D scatter points in different tabs respectively. We've got the 3D points plotting idea from the community. However we found no relative topics about 2D scatter plotting. Can anyone help on it?

Your advice is deeply appreciated.
Regards,
DR, Ling

[marcomusy]

Hi, you can find many examples of scatter plots in examples/pyplot and in examples.other/qt_*.py e.g.:

import sys
from PyQt5 import QtCore, QtWidgets
from vtk.qt.QVTKRenderWindowInteractor import QVTKRenderWindowInteractor
from vedo import Mesh, dataurl, Plotter
from vedo.pyplot import np, histogram

try:
    _encoding = QtWidgets.QApplication.UnicodeUTF8
    def _translate(context, text, disambig):
        return QtWidgets.QApplication.translate(context, text, disambig, _encoding)
except AttributeError:
    def _translate(context, text, disambig):
        return QtWidgets.QApplication.translate(context, text, disambig)


class Ui_MainWindow(object):

    def setupUi(self, MainWindow):
        MainWindow.setObjectName("MainWindow")
        MainWindow.resize(800, 600)

        self.centralwidget = QtWidgets.QWidget(MainWindow)
        self.centralwidget.setObjectName("centralwidget")

        self.tab1 = QtWidgets.QWidget()
        self.tab1.setObjectName("tab1")
        self.gridLayout1 = QtWidgets.QGridLayout(self.tab1)

        self.vtkLayout1 = QtWidgets.QVBoxLayout()
        self.vtkLayout1.setObjectName("vtkLayout1")
        self.gridLayout1.addLayout(self.vtkLayout1, 0, 0, 1, 1)

        self.tab2 = QtWidgets.QWidget()

        self.tabWidget = QtWidgets.QTabWidget(self.centralwidget)
        self.tabWidget.addTab(self.tab1, "tab1")
        self.tabWidget.addTab(self.tab2, "tab2")

        self.verticalLayout = QtWidgets.QVBoxLayout()
        self.verticalLayout.addWidget(self.tabWidget)

        self.gridLayout2 = QtWidgets.QGridLayout(self.centralwidget)
        self.gridLayout2.addLayout(self.verticalLayout, 0, 0, 1, 1)

        MainWindow.setCentralWidget(self.centralwidget)
        self.menubar = QtWidgets.QMenuBar(MainWindow)
        self.menubar.setGeometry(QtCore.QRect(0, 0, 800, 31))

        MainWindow.setMenuBar(self.menubar)
        self.statusbar = QtWidgets.QStatusBar(MainWindow)
        MainWindow.setStatusBar(self.statusbar)

        MainWindow.setWindowTitle(_translate("MainWindow", "MainWindow", None))
        self.tabWidget.setTabText(
            self.tabWidget.indexOf(self.tab1), _translate("MainWindow", "Tab 1", None)
        )
        self.tabWidget.setTabText(
            self.tabWidget.indexOf(self.tab2), _translate("MainWindow", "Tab 2", None)
        )

        self.tabWidget.setCurrentIndex(0)
        QtCore.QMetaObject.connectSlotsByName(MainWindow)


class MainWindow(QtWidgets.QMainWindow, Ui_MainWindow):

    def __init__(self, parent=None):
        QtWidgets.QMainWindow.__init__(self, parent)
        self.setupUi(self)

        self.vtkWidget1 = QVTKRenderWindowInteractor(self)
        self.vtkLayout1.addWidget(self.vtkWidget1)

        self.plt1 = Plotter(qt_widget=self.vtkWidget1, axes=1)
        self.id2 = self.plt1.add_callback("key press", self.onKeypress)
        self.plt1 += Mesh(dataurl+'shark.ply').c('cyan')
        self.plt1.show()

        self.vtkWidget2 = QVTKRenderWindowInteractor(self)
        self.verticalLayout.addWidget(self.vtkWidget2)
        self.plt2 = Plotter(qt_widget=self.vtkWidget2)
        data = np.random.randn(1000)
        self.plt2 += histogram(data, aspect=16/9)
        self.plt2.show(zoom=2, mode='image')

    def onClose(self):
        self.vtkWidget1.close()

    def onKeypress(self, evt):
        print("You have pressed key:", evt.keypress)
        if evt.keypress=='q':
            sys.exit(0)


if __name__ == "__main__":
    app = QtWidgets.QApplication(sys.argv)
    window = MainWindow()
    app.aboutToQuit.connect(window.onClose)
    window.show()
    sys.exit(app.exec_())

[DRLing2021]

Thank you for the example, Marcomusy!

I need to make my questions more specific that I am looking for 2D scatter point object in Vedo. I did see the 2D histogram examples but not know any 2D point or 2D line object. So the plan is to plot the 3D point cloud, which we can do it correctly, and 2D scatter points, which we don't know how to, at the same Qt5 UI.

Regards!

[marcomusy]

Hi these are other 2 examples which include a scatter plot:

import sys
from PyQt5 import QtCore, QtWidgets
from vtk.qt.QVTKRenderWindowInteractor import QVTKRenderWindowInteractor
from vedo import Mesh, dataurl, Plotter, Points, precision, Axes
from vedo.pyplot import np, histogram, plot

try:
    _encoding = QtWidgets.QApplication.UnicodeUTF8
    def _translate(context, text, disambig):
        return QtWidgets.QApplication.translate(context, text, disambig, _encoding)
except AttributeError:
    def _translate(context, text, disambig):
        return QtWidgets.QApplication.translate(context, text, disambig)


class Ui_MainWindow(object):

    def setupUi(self, MainWindow):
        MainWindow.setObjectName("MainWindow")
        MainWindow.resize(800, 600)

        self.centralwidget = QtWidgets.QWidget(MainWindow)
        self.centralwidget.setObjectName("centralwidget")

        self.tab1 = QtWidgets.QWidget()
        self.tab1.setObjectName("tab1")
        self.gridLayout1 = QtWidgets.QGridLayout(self.tab1)

        self.vtkLayout1 = QtWidgets.QVBoxLayout()
        self.vtkLayout1.setObjectName("vtkLayout1")
        self.gridLayout1.addLayout(self.vtkLayout1, 0, 0, 1, 1)

        self.tab2 = QtWidgets.QWidget()

        self.tabWidget = QtWidgets.QTabWidget(self.centralwidget)
        self.tabWidget.addTab(self.tab1, "tab1")
        self.tabWidget.addTab(self.tab2, "tab2")

        self.verticalLayout = QtWidgets.QVBoxLayout()
        self.verticalLayout.addWidget(self.tabWidget)

        self.gridLayout2 = QtWidgets.QGridLayout(self.centralwidget)
        self.gridLayout2.addLayout(self.verticalLayout, 0, 0, 1, 1)

        MainWindow.setCentralWidget(self.centralwidget)
        self.menubar = QtWidgets.QMenuBar(MainWindow)
        self.menubar.setGeometry(QtCore.QRect(0, 0, 800, 31))

        MainWindow.setMenuBar(self.menubar)
        self.statusbar = QtWidgets.QStatusBar(MainWindow)
        MainWindow.setStatusBar(self.statusbar)

        MainWindow.setWindowTitle(_translate("MainWindow", "MainWindow", None))
        self.tabWidget.setTabText(
            self.tabWidget.indexOf(self.tab1), _translate("MainWindow", "Tab 1", None)
        )
        self.tabWidget.setTabText(
            self.tabWidget.indexOf(self.tab2), _translate("MainWindow", "Tab 2", None)
        )

        self.tabWidget.setCurrentIndex(0)
        QtCore.QMetaObject.connectSlotsByName(MainWindow)


class MainWindow(QtWidgets.QMainWindow, Ui_MainWindow):

    def __init__(self, parent=None):
        QtWidgets.QMainWindow.__init__(self, parent)
        self.setupUi(self)

        self.vtkWidget1 = QVTKRenderWindowInteractor(self)
        self.vtkLayout1.addWidget(self.vtkWidget1)

        self.plt1 = Plotter(qt_widget=self.vtkWidget1, axes=1)
        self.id2 = self.plt1.add_callback("key press", self.onKeypress)
        self.plt1 += Mesh(dataurl+'shark.ply').c('cyan')
        self.plt1.show()

        self.vtkWidget2 = QVTKRenderWindowInteractor(self)
        self.verticalLayout.addWidget(self.vtkWidget2)
        self.plt2 = Plotter(qt_widget=self.vtkWidget2)

        #####################################################
        # first example using a formatted Figure
        x = np.random.randn(100) + 10
        y = np.random.randn(100) * 20 + 20
        fig = plot(
            x, y,
            lw=0,         # do not join points with lines
            xtitle="variable x",
            ytitle="variable y",
            marker="*",   # marker style
            mc="dr",      # marker color
            aspect=16/9,  # aspect ratio
        )
        self.plt2 += fig

        #####################################################
        # # second example using Points and a density estimator
        # n = 10000
        # p = np.random.normal(0, 0.3, (n,2))
        # p[:int(n*1/3) ] += [1.0, 0.0] # shift 1/3 of the points along x by 1
        # p[ int(n*2/3):] += [1.7, 0.4]

        # # create the point cloud
        # pts = Points(p).color('k').alpha(0.4)

        # vol = pts.density().c('Paired_r') # returns a 1-voxel Volume
        # r = precision(vol.info['radius'], 2) # retrieve automatic radius value
        # vol.add_scalarbar3d(title='Density (counts in r_search ='+r+')', c='k', italic=1)

        # self.plt2 += [pts, vol, Axes(pts)]


        ##################################################### show
        self.plt2.show(zoom=1.8, mode='image')

    def onClose(self):
        self.vtkWidget1.close()

    def onKeypress(self, evt):
        print("You have pressed key:", evt.keypress)
        if evt.keypress=='q':
            sys.exit(0)


if __name__ == "__main__":
    app = QtWidgets.QApplication(sys.argv)
    window = MainWindow()
    app.aboutToQuit.connect(window.onClose)
    window.show()
    sys.exit(app.exec_())

Let me know if this answer your question.

You can search examples by keyword with eg

vedo --search scatter

and run them with

vedo --run scatter2

Hope this helps!

[DRLing2021]

They are exactly what I am looking for!

Thank you very much Marco!