如何使用tkinter正确显示熊猫数据框?

时间:2019-09-07 01:01:37

标签: python dataframe tkinter

我对python还是很陌生,我正在尝试创建一个工具来显示文件夹中Excel工作簿所有工作表的行数和列数。我希望使用tkinter将数据框显示为最终结果,但是由于数据框的最后两列出现在新行上,因此显示不正确。我想知道如何解决这个问题。我尝试使用PyQT5,但是这一直使我的内核崩溃,并且我尝试使用Treeviews,但是我不知道如何正确地将此数据帧写入Treeview。下面是我当前的代码:

import pandas as pd
import tkinter as tk
import glob 
import os
import xlrd

def folder_row_count():
    folder_path = f_path_entry.get()
    file_extension = file_ext_var.get()
    window = tk.Tk()
    t1 = tk.Text(window)
    t1.grid()
    if file_extension == "xlsx":
        filenames = []
        sheetnames = []
        sheetrows = []
        sheetcols = []
        for fname in  glob.glob(os.path.join(folder_path, f"*.{file_extension}")):
            wb = xlrd.open_workbook(fname)
            filename = []
            sheetname = []
            sheetrow = []
            sheetcol = []
            for sheet in wb.sheets():
                filename.append(os.path.basename(fname))
                sheetname.append(sheet.name)
                sheetrow.append(sheet.nrows)
                sheetcol.append(sheet.ncols)
            filenames.append(filename)
            sheetnames.append(sheetname)
            sheetrows.append(sheetrow)
            sheetcols.append(sheetcol)
        flat_filenames = [item for filename in filenames for item in filename]
        flat_sheetnames = [item for sheetname in sheetnames for item in sheetname]
        flat_sheetrows = [item for sheetrow in sheetrows for item in sheetrow]
        flat_sheetcols = [item for sheetcol in sheetcols for item in sheetcol]
    df = pd.DataFrame({'File Name': flat_filenames,
                       'Sheet Name': flat_sheetnames,
                       'Number Of Rows': flat_sheetrows,
                       'Number Of Columns': flat_sheetcols
                        })
    main_df = df.append(df.sum(numeric_only = True).rename('Total'))
    t1.insert(tk.END, main_df)
    window.mainloop()

file_ext_list = ["xlsx"]

window = tk.Tk()

window.title("Row Counter")

tk.Label(window, text = "Choose File Type:").grid(row = 1, column = 0)

file_ext_var = tk.StringVar(window)
file_ext_dd = tk.OptionMenu(window, file_ext_var, *file_ext_list)
file_ext_dd.config(width = 10)
file_ext_dd.grid(row = 1, column = 1)

tk.Label(window, text = "Folder Path:").grid(row = 2, column = 0)

f_path_entry = tk.Entry(window)
f_path_entry.grid(row = 2, column = 1)

tk.Button(window, text = "Count Rows", command = folder_row_count).grid(row = 4, column = 1)

window.mainloop()

第二,对于我如何改进此代码并使其更有效的任何评论,我将不胜感激。

谢谢。

1 个答案:

答案 0 :(得分:0)

您只需要通过df遍历iterrows并将它们插入Treeview中即可。下面是一个基本示例:

import tkinter as tk
from tkinter import ttk
import pandas as pd

root = tk.Tk()

sample = {"File Name":[f"file_{i}" for i in range(5)],
          'Sheet Name': [f"sheet_{i}" for i in range(5)],
          'Number Of Rows': [f"row_{i}" for i in range(5)],
          'Number Of Columns': [f"col_{i}" for i in range(5)]
          }
df = pd.DataFrame(sample)
cols = list(df.columns)

tree = ttk.Treeview(root)
tree.pack()
tree["columns"] = cols
for i in cols:
    tree.column(i, anchor="w")
    tree.heading(i, text=i, anchor='w')

for index, row in df.iterrows():
    tree.insert("",0,text=index,values=list(row))

root.mainloop()

我还看到您正在使用xlrd来先阅读您的excel,然后再将其转换为Dataframe。为什么不改用pandas.read_excel