Closes Bears-R-Us#3116 remove DataFrame._columns

ajpotts · Apr 29, 2024 · 6993791 · 6993791
1 parent 1fcbbd4
commit 6993791
Showing 1 changed file with 3 additions and 8 deletions.
diff --git a/arkouda/dataframe.py b/arkouda/dataframe.py
@@ -685,9 +685,7 @@ def __init__(self, initialdata=None, index=None, columns=None):
                     if len(sizes) > 1:
                         raise ValueError("Input arrays must have equal size.")
                     self._empty = False
-                    UserDict.__setitem__(self, key, val)
-                    # Update the column index
-                    self._columns.append(key)
+                    self[key] = val
 
             # Initial data is a list of arkouda arrays
             elif isinstance(initialdata, list):
@@ -711,9 +709,7 @@ def __init__(self, initialdata=None, index=None, columns=None):
                     if len(sizes) > 1:
                         raise ValueError("Input arrays must have equal size.")
                     self._empty = False
-                    UserDict.__setitem__(self, key, col)
-                    # Update the column index
-                    self._columns.append(key)
+                    self[key] = col
 
             # Initial data is invalid.
             else:
@@ -778,8 +774,7 @@ def __getitem__(self, key):
                 raise TypeError("Invalid selector: too many types in list.")
             if isinstance(key[0], str):
                 for k in key:
-                    result.data[k] = UserDict.__getitem__(self, k)
-                    result._columns.append(k)
+                    result[k] = self[k]
                 result._empty = False
                 result._set_index(self.index)  # column lens remain the same. Copy the indexing
                 return result