'DataFrame column number'

def columnCount(self, index=QModelIndex()):

    if (len(self.df.shape) == 1):
        return 2
    elif (self.total_cols <= self.cols_loaded):
        return (self.total_cols + 1)
    else:
        return (self.cols_loaded + 1)
