Возврат объекта виджета кнопки включения Python - PullRequest
Новая
       23

Возврат объекта виджета кнопки включения Python

0 голосов
/ 21 февраля 2019

Я пытаюсь создать селектор файлов / данных в ноутбуках Jupyter с Python.Идея состоит в том, что я выбираю некоторые файлы и каналы данных в файлах с помощью виджета множественного выбора, а затем с помощью кнопки возвращаю фрейм данных.

Как получить доступ к df_object? 

#stack example
from ipywidgets import widgets
from IPython.display import display
from IPython.display import clear_output
import pandas as pd
import numpy as np
filenames = ["file1", "file2"]
file_dict = {
    "file1":pd.DataFrame(np.arange(5)),
    "file2":pd.DataFrame(np.arange(10,15))
}

def data_selection():
    sel_file = widgets.SelectMultiple(description="Files",
    options=filenames)
    display(sel_file)

    button = widgets.Button(description="OK")
    display(button)            

    def on_button_clicked(button):
        clear_output(wait=True) #clears the previous output
        display(sel_file) #displays new selection window
        display(button) #displays new button
        for f in sel_file.value:
            print (f)
            display (file_dict[f])
            #global df_object #would be a solution but not recommended for sure
            df_object = file_dict[f]
            return df_object #doesn't work
    button.on_click(on_button_clicked)

data_selection()

1 Ответ

0 голосов
/ 28 февраля 2019

Вы действительно должны использовать класс для этого, а затем определить все свои функции как действующие на экземпляр этого класса.Не все из них должны быть общедоступными.Вы также можете сохранить df_objects в отдельном атрибуте, таком как словарь, и получить доступ к словарю с помощью отдельной функции.Проверьте код ниже: 

class foo(object):

    def __init__(self, file1, file2):
        self.filenames = [file1, file2]
        self.file_dict = {
                file1:pd.DataFrame(np.arange(5)),
                file2:pd.DataFrame(np.arange(10,15))
            }

    def _create_widgets(self):
        self.sel_file = widgets.SelectMultiple(description='Files',
                                               options=self.filenames,
                                               value=[self.filenames[0]],
                                              )
        self.button = widgets.Button(description="OK")
        self.button.on_click(self._on_button_clicked)

    def _on_button_clicked(self, change):
        self.out.clear_output()
        self.df_objects = {}
        with self.out:
            for f in self.sel_file.value:
                print(f)
                display(self.file_dict[f])
                self.df_objects[f] = self.file_dict[f]

    def display_widgets(self):
        self._create_widgets()
        self.out = widgets.Output()  # this is the output widget in which the df is displayed
        display(widgets.VBox(
                            [
                                self.sel_file,
                                self.button,
                                self.out
                            ]
                        )
               )

    def get_df_objects(self):
        return self.df_objects

Затем вы можете создавать экземпляры и отображать виджеты следующим образом: 

something = foo('a', 'b')
something.display_widgets()

something.get_df_objects() вернет словарь с требуемым 'file:dataframe_of_file'пары ключ-значение.

Надеюсь, это поможет:)

...