Interpret astype(object) as 'cast to an array of numpy arrays'

Author: frreiss

text_extensions_for_pandas/array/tensor.py

@@ -482,6 +482,11 @@ def astype(self, dtype, copy=True):
                 return dtype.construct_array_type()._from_sequence(values, copy=False)
             else:
                 return values
+        elif pd.api.types.is_object_dtype(dtype):
+            # Interpret astype(object) as "cast to an array of numpy arrays"
+            values = np.empty(len(self), dtype=object)
+            for i in range(len(self)):
+                values[i] = self._tensor[i]
         else:
             values = self._tensor.astype(dtype, copy=copy)
         return values