DataFrame的自定义ContextManager

Question

我在Python3中创建了一个上下文管理器来修改和管理熊猫的DataFrame。

我有三个函数，delete，update和insert。

所以我的问题是，我如何才能将数据帧的结果传回给with语句之外的用户？

```javascript

df = pd.DataFrame()

使用编辑器(Df)作为编辑器：

for idx, row in editor:

    row['COLUMN'] = "RANDOM VALUE"

    editor.update(idx, row)

df“‘COLUMN”#仍具有原始值

如何从ContextManager取回更新后的值？

Answer 1

您的上下文管理器可能如下所示：

class Editor():
    def __init__(self, df):
        self.df = df

    def __enter__(self):
        return self.df

    def __exit__(self, *args):
        pass

    def update(self):
        pass

因此，获得修改后的对象，您可以在for循环完成后立即执行类似df = editor.df的操作。

不过，我不确定您从上下文管理器中获得了什么好处。