From 44fe52cee636e268a31c42db8131b0c99d63521c Mon Sep 17 00:00:00 2001 From: AChenQ Date: Mon, 27 Jun 2022 16:37:27 +0800 Subject: [PATCH] feat(dataframe): support "__delitem__" method in "DataFrame" --- graviti/dataframe/frame.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/graviti/dataframe/frame.py b/graviti/dataframe/frame.py index 7aad4bb2..500b4e1b 100644 --- a/graviti/dataframe/frame.py +++ b/graviti/dataframe/frame.py @@ -150,6 +150,11 @@ def __setitem__(self, key: str, value: Union[Iterable[Any], _C]) -> None: dataframe._record_key = self._record_key self.operations.extend((UpdateSchema(self.schema), UpdateData(dataframe))) + def __delitem__(self, key: str) -> None: + del self._columns[key] + del self.schema[key] + # TODO: Need add delete operations. + def __repr__(self) -> str: flatten_header, flatten_data = self._flatten() header = self._get_repr_header(flatten_header)