Merge pull request pandas-dev#5013 from alefnula/qtpandas

ENH: PySide support for qtpandas.

Author: jreback

doc/source/release.rst

@@ -77,6 +77,7 @@ Experimental Features
     (:issue:`4897`).
   - Add msgpack support via ``pd.read_msgpack()`` and ``pd.to_msgpack()`` / ``df.to_msgpack()`` for serialization
     of arbitrary pandas (and python objects) in a lightweight portable binary format (:issue:`686`)
+  - Added PySide support for the qtpandas DataFrameModel and DataFrameWidget.
 
 Improvements to existing features
 ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

doc/source/v0.13.0.txt

@@ -604,6 +604,8 @@ Experimental
 
      os.remove('foo.msg')
 
+- Added PySide support for the qtpandas DataFrameModel and DataFrameWidget.
+
 .. _whatsnew_0130.refactoring:
 
 Internal Refactoring

doc/source/visualization.rst

@@ -594,3 +594,79 @@ Andrews curves charts:
 
    @savefig andrews_curve_winter.png
    andrews_curves(data, 'Name', colormap='winter')
+
+
+****************************************
+Visualizing your data in Qt applications
+****************************************
+
+There is an experimental support for visualizing DataFrames in PyQt4 and PySide
+applications. At the moment you can display and edit the values of the cells
+in the DataFrame. Qt will take care of displaying just the portion of the
+DataFrame that is currently visible and the edits will be immediately saved to
+the underlying DataFrame
+
+To demonstrate this we will create a simple PySide application that will switch
+between two editable DataFrames. For this will use the ``DataFrameModel`` class
+that handles the access to the DataFrame, and the ``DataFrameWidget``, which is
+just a thin layer around the ``QTableView``.
+
+.. code-block:: python
+
+	import numpy as np
+	import pandas as pd
+	from pandas.sandbox.qtpandas import DataFrameModel, DataFrameWidget 
+	from PySide import QtGui, QtCore
+	
+	# Or if you use PyQt4:
+	# from PyQt4 import QtGui, QtCore
+	
+	class MainWidget(QtGui.QWidget):
+	    def __init__(self, parent=None):
+	        super(MainWidget, self).__init__(parent)
+	        
+	        # Create two DataFrames
+	        self.df1 = pd.DataFrame(np.arange(9).reshape(3, 3),
+	                                columns=['foo', 'bar', 'baz'])
+	        self.df2 = pd.DataFrame({
+	                'int': [1, 2, 3],
+	                'float': [1.5, 2.5, 3.5],
+	                'string': ['a', 'b', 'c'],
+	                'nan': [np.nan, np.nan, np.nan]
+	            }, index=['AAA', 'BBB', 'CCC'],
+	            columns=['int', 'float', 'string', 'nan'])
+	
+	        # Create the widget and set the first DataFrame
+	        self.widget = DataFrameWidget(self.df1)
+	        
+	        # Create the buttons for changing DataFrames
+	        self.button_first = QtGui.QPushButton('First')
+	        self.button_first.clicked.connect(self.on_first_click)
+	        self.button_second = QtGui.QPushButton('Second')
+	        self.button_second.clicked.connect(self.on_second_click)
+	        
+	        # Set the layout
+	        vbox = QtGui.QVBoxLayout()
+	        vbox.addWidget(self.widget)
+	        hbox = QtGui.QHBoxLayout()
+	        hbox.addWidget(self.button_first)
+	        hbox.addWidget(self.button_second)
+	        vbox.addLayout(hbox)
+	        self.setLayout(vbox)
+	    
+	    def on_first_click(self):
+	    	'''Sets the first DataFrame'''
+	        self.widget.setDataFrame(self.df1)
+	    
+	    def on_second_click(self):
+	    	'''Sets the second DataFrame'''
+	        self.widget.setDataFrame(self.df2)
+	
+	if __name__ == '__main__':
+	    import sys
+	    
+	    # Initialize the application
+	    app = QtGui.QApplication(sys.argv)
+	    mw = MainWidget()
+	    mw.show()
+	    app.exec_()

pandas/sandbox/qtpandas.py

@@ -3,10 +3,15 @@
 
 @author: Jev Kuznetsov
 '''
-from PyQt4.QtCore import (
-    QAbstractTableModel, Qt, QVariant, QModelIndex, SIGNAL)
-from PyQt4.QtGui import (
-    QApplication, QDialog, QVBoxLayout, QTableView, QWidget)
+try:
+    from PyQt4.QtCore import QAbstractTableModel, Qt, QVariant, QModelIndex
+    from PyQt4.QtGui import (
+        QApplication, QDialog, QVBoxLayout, QTableView, QWidget)
+except ImportError:
+    from PySide.QtCore import QAbstractTableModel, Qt, QModelIndex
+    from PySide.QtGui import (
+        QApplication, QDialog, QVBoxLayout, QTableView, QWidget)
+    QVariant = lambda value=None: value
 
 from pandas import DataFrame, Index
 
@@ -57,9 +62,17 @@ def flags(self, index):
             return flags
 
     def setData(self, index, value, role):
-        self.df.set_value(self.df.index[index.row()],
-                          self.df.columns[index.column()],
-                          value.toPyObject())
+        row = self.df.index[index.row()]
+        col = self.df.columns[index.column()]
+        if hasattr(value, 'toPyObject'):
+            # PyQt4 gets a QVariant
+            value = value.toPyObject()
+        else:
+            # PySide gets an unicode
+            dtype = self.df[col].dtype
+            if dtype != object:
+                value = None if value == '' else dtype.type(value)
+        self.df.set_value(row, col, value)
         return True
 
     def rowCount(self, index=QModelIndex()):
@@ -75,17 +88,18 @@ def __init__(self, dataFrame, parent=None):
         super(DataFrameWidget, self).__init__(parent)
 
         self.dataModel = DataFrameModel()
-        self.dataModel.setDataFrame(dataFrame)
-
         self.dataTable = QTableView()
         self.dataTable.setModel(self.dataModel)
-        self.dataModel.signalUpdate()
 
         layout = QVBoxLayout()
         layout.addWidget(self.dataTable)
         self.setLayout(layout)
+        # Set DataFrame
+        self.setDataFrame(dataFrame)
 
-    def resizeColumnsToContents(self):
+    def setDataFrame(self, dataFrame):
+        self.dataModel.setDataFrame(dataFrame)
+        self.dataModel.signalUpdate()
         self.dataTable.resizeColumnsToContents()
 
 #-----------------stand alone test code