我如何从QTableWidget检索数据到Dataframe?

我有一个处于可编辑模式的QTableWidget,用户在其中输入整数输入,如何生成在此表中输入的数据列表,以便对其执行操作,这是我的手动代码:

def dataframe_generation_from_table(self,table):
    number_of_rows = table.rowCount()
    number_of_columns = table.columnCount()

    tmp_df = pd.DataFrame({ 'Date' : [] , str(self.final_lvl_of_analysis) :[],  'Value': []}) 

    for i in range(0,number_of_rows):
        for j in range(0,number_of_columns):
            tmp_item = table.item(i,j)
            tmp_df2 = pd.DataFrame( { 'Date' : [pd.to_datetime(table.horizontalHeaderItem(j).data())] , str(self.final_lvl_of_analysis) :[ str(table.verticalHeaderItem(i).data())], 'Value': [float(tmp_item.data(0))]})
            print tmp_df2
            tmp_df.update(tmp_df2, join = 'left', overwrite = False)

    return tmp_df

另外,我正在使用以下代码来生成QTableWidget:

    self.pd_table = QtGui.QTableWidget(self.groupBox_19)
    self.pd_table.setObjectName(_fromUtf8("pd_table"))
    self.pd_table.setColumnCount(0)
    self.pd_table.setRowCount(0)

我的规格是:pandas 0.18.1,PyQt 4和Python 2.7

解决方法:

我认为您在更新/联接方面使它过于复杂.最简单的方法是先创建完整尺寸的DataFrame(用NaN填充),然后将数据分配给此:

def dataframe_generation_from_table(self,table):
    number_of_rows = table.rowCount()
    number_of_columns = table.columnCount()

    tmp_df = pd.DataFrame( 
                columns=['Date', str(self.final_lvl_of_analysis), 'Value'], # Fill columnets
                index=range(number_of_rows) # Fill rows
                ) 

    for i in range(number_of_rows):
        for j in range(number_of_columns):
            tmp_df.ix[i, j] = table.item(i, j).data()

    return tmp_df

上面的代码通过数字索引将数据分配到其位置,因此QtTableWidget中的位置1,1将最终在DataFrame中的1,1处结束.这样,您在移动数据时无需担心列标题.如果要更改列名,则可以在创建DataFrame时执行此操作,更改传递到columns =参数的值.

如果要将列更改为DateTime格式,则应该可以在循环后通过以下单个操作执行此操作:

tmp_df['Date'] = pd.to_datetime( tmp_df['Date'] )
上一篇:如何从pyside中的滑块获取值?


下一篇:Codeforces Round #575 (Div. 3) A - Three Piles of Candies