根据原有dataframe生成子dataframe

可以使用 pandas 中的 loc 或者 iloc 方法来基于原有 dataframe 生成子 dataframe。

loc 方法是基于行标签和列标签进行索引，示例如下：

import pandas as pd

# 生成原有 dataframe
df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6], 'C': [7, 8, 9]}, index=['a', 'b', 'c'])

# 使用 loc 方法生成子 dataframe
sub_df = df.loc[['a', 'c'], ['A', 'C']]
print(sub_df)

输出结果为：

   A  C
a  1  7
c  3  9

iloc 方法是基于行索引和列索引进行索引，示例如下：

import pandas as pd

# 生成原有 dataframe
df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6], 'C': [7, 8, 9]}, index=['a', 'b', 'c'])

# 使用 iloc 方法生成子 dataframe
sub_df = df.iloc[[0, 2], [0, 2]]
print(sub_df)

输出结果为：

   A  C
a  1  7
c  3  9

相关问题

用python生成一个GUi，具备以下功能： 1.界面内包含数据文件读取、表格中显示读取的数据以及绘图区域； 2.选择数据文件后，在表格中显示对应的数据，支持表格中数据的编辑和报存。 3.通过点击按钮可新增绘图子页面。

下面是一个简单的Python GUI程序，实现了以上三个功能：

import tkinter as tk
from tkinter import filedialog
import pandas as pd
import matplotlib.pyplot as plt

class App(tk.Tk):
    def __init__(self):
        super().__init__()
        self.title("数据分析")
        self.geometry("800x600")
        self.data = None

        # 创建菜单栏
        menubar = tk.Menu(self)
        filemenu = tk.Menu(menubar, tearoff=0)
        filemenu.add_command(label="打开", command=self.open_file)
        filemenu.add_command(label="保存", command=self.save_file)
        menubar.add_cascade(label="文件", menu=filemenu)
        self.config(menu=menubar)

        # 创建表格
        self.table = tk.Frame(self)
        self.table.pack(side="top", fill="both", expand=True)
        self.table_data = pd.DataFrame(columns=["列1", "列2", "列3"])
        self.table_view = pdgui.Table(self.table, data=self.table_data)
        self.table_view.pack(side="left", fill="both", expand=True)

        # 创建绘图区域
        self.figure = plt.figure(figsize=(8, 6), dpi=100)
        self.canvas = FigureCanvasTkAgg(self.figure, master=self)
        self.canvas.draw()
        self.canvas.get_tk_widget().pack(side="top", fill="both", expand=True)

        # 创建按钮
        self.add_plot_button = tk.Button(self, text="新增绘图子页面", command=self.add_plot_page)
        self.add_plot_button.pack(side="bottom")

    def open_file(self):
        filename = filedialog.askopenfilename()
        if filename:
            self.data = pd.read_csv(filename)
            self.table_view.update_data(self.data)

    def save_file(self):
        if self.data is None:
            return
        filename = filedialog.asksaveasfilename()
        if filename:
            self.data.to_csv(filename, index=False)

    def add_plot_page(self):
        plot_page = pdgui.PlotPage(self.figure)
        plot_page.show()

class Table(tk.Frame):
    def __init__(self, master, data=None):
        super().__init__(master)
        self.master = master
        self.data = data
        self.create_widgets()

    def create_widgets(self):
        if self.data is None:
            return
        self.treeview = ttk.Treeview(self, selectmode="browse")
        self.treeview.pack(side="left", fill="both", expand=True)
        self.treeview["columns"] = list(self.data.columns)
        for col in self.data.columns:
            self.treeview.column(col, width=100, anchor="center")
            self.treeview.heading(col, text=col)
        for i, row in self.data.iterrows():
            self.treeview.insert("", "end", values=row.tolist())

    def update_data(self, data):
        self.data = data
        self.treeview.delete(*self.treeview.get_children())
        self.create_widgets()

class PlotPage():
    def __init__(self, figure):
        self.figure = figure
        self.ax = self.figure.add_subplot(111)

    def show(self):
        self.ax.plot([1, 2, 3], [4, 5, 6])
        self.figure.canvas.draw()

if __name__ == "__main__":
    app = App()
    app.mainloop()

在这个程序中，我们使用了Tkinter库来创建GUI界面，同时使用了Pandas和Matplotlib库来处理数据和绘图。在GUI界面中，我们使用了菜单栏、表格和绘图区域来展示数据和绘图结果，同时添加了一个按钮，可以新增绘图子页面。具体实现方法如下：

- 创建App类，继承自Tk的窗口类。在创建窗口时，设置了标题和大小，并定义了一个成员变量data，用于存储读取的数据。
- 在App类的构造函数中，创建菜单栏，包含“打开”和“保存”两个选项。同时创建表格和绘图区域，并添加一个按钮。
- 实现open_file方法，用于打开数据文件。使用filedialog库中的askopenfilename函数获取文件名，读取文件内容，并将读取的数据更新到表格中。
- 实现save_file方法，用于保存数据文件。如果没有数据需要保存，直接返回。使用filedialog库中的asksaveasfilename函数获取保存文件名，将数据保存到文件中。
- 实现add_plot_page方法，用于新增绘图子页面。创建一个PlotPage对象，将其添加到figure对象中，并调用show方法显示绘图结果。
- 创建Table类，继承自Tk的Frame类。在创建表格时，需要传入数据，并使用ttk库中的Treeview对象来展示数据。使用iterrows方法遍历数据，将每行数据插入到Treeview中。
- 实现update_data方法，用于更新数据。首先删除原有的Treeview对象，然后重新创建新的Treeview对象来展示新的数据。
- 创建PlotPage类，用于展示绘图结果。在构造函数中，传入figure对象，并创建一个Axes对象。在show方法中，绘制一个简单的折线图，并调用figure对象的canvas.draw方法来显示绘图结果。

上述程序只是一个简单的示例，可以根据具体需求进行修改和扩展。例如，可以添加数据分析功能、交互式绘图功能等。