Add property: DataFrame.ndim, Index.ndim, MultiIndex.ndim

databricks/koalas/frame.py

@@ -377,6 +377,29 @@ def __init__(self, data=None, index=None, columns=None, dtype=None, copy=False):
     def _sdf(self) -> spark.DataFrame:
         return self._internal.sdf
 
+    @property
+    def ndim(self):
+        """
+        Return an int representing the number of array dimensions.
+
+        Return 1 if Series. Otherwise return 2 if DataFrame.
+
+        Examples
+        --------
+
+        >>> df = ks.DataFrame([[1, 2], [4, 5], [7, 8]],
+        ...                   index=['cobra', 'viper', None],
+        ...                   columns=['max_speed', 'shield'])
+        >>> df
+               max_speed  shield
+        cobra          1       2
+        viper          4       5
+        NaN            7       8
+        >>> df.ndim
+        2
+        """
+        return 2
+
     def _reduce_for_stat_function(self, sfun, name, axis=None, numeric_only=False):
         """
         Applies sfun to each column and returns a pd.Series where the number of rows equal the

databricks/koalas/indexes.py

@@ -151,6 +151,27 @@ def names(self, names: List[Union[str, Tuple[str, ...]]]) -> None:
         names = [name if isinstance(name, tuple) else (name,) for name in names]
         self._kdf._internal = internal.copy(index_map=list(zip(internal.index_columns, names)))
 
+    @property
+    def ndim(self):
+        """
+        Number of dimensions of the underlying data, by definition 1.
+
+        Examples
+        --------
+        >>> s = ks.Series([None, 1, 2, 3, 4], index=[4, 5, 2, 1, 8])
+        >>> s.index.ndim
+        1
+
+        >>> midx = pd.MultiIndex([['lama', 'cow', 'falcon'],
+        ...                       ['speed', 'weight', 'length']],
+        ...                      [[0, 0, 0, 1, 1, 1, 2, 2, 2],
+        ...                       [1, 1, 1, 1, 1, 2, 1, 2, 2]])
+        >>> s = ks.Series([45, 200, 1.2, 30, 250, 1.5, 320, 1, 0.3], index=midx)
+        >>> s.index.ndim
+        1
+        """
+        return 1
+
     def rename(self, name: Union[str, Tuple[str, ...]], inplace: bool = False):
         """
         Alter Index name.

databricks/koalas/missing/frame.py

@@ -32,7 +32,6 @@ class _MissingPandasLikeDataFrame(object):
     # Properties
     axes = unsupported_property('axes')
     iat = unsupported_property('iat')
-    ndim = unsupported_property('ndim')
 
     # Deprecated properties
     blocks = unsupported_property('blocks', deprecated=True)

databricks/koalas/missing/indexes.py

@@ -33,7 +33,6 @@ class _MissingPandasLikeIndex(object):
     T = unsupported_property('T')
     has_duplicates = unsupported_property('has_duplicates')
     nbytes = unsupported_property('nbytes')
-    ndim = unsupported_property('ndim')
     nlevels = unsupported_property('nlevels')
     shape = unsupported_property('shape')
 
@@ -134,7 +133,6 @@ class _MissingPandasLikeMultiIndex(object):
     is_all_dates = unsupported_property('is_all_dates')
     levels = unsupported_property('levels')
     levshape = unsupported_property('levshape')
-    ndim = unsupported_property('ndim')
     nlevels = unsupported_property('nlevels')
     shape = unsupported_property('shape')
 

docs/source/reference/frame.rst

@@ -27,6 +27,7 @@ Attributes and underlying data
 
    DataFrame.dtypes
    DataFrame.shape
+   DataFrame.ndim
    DataFrame.size
    DataFrame.select_dtypes
 

docs/source/reference/indexing.rst

@@ -21,6 +21,7 @@ Properties
    Index.dtype
    Index.name
    Index.names
+   Index.ndim
    Index.empty
 
 Modifying and computations
@@ -76,6 +77,7 @@ MultiIndex Properties
    :toctree: api/
 
    MultiIndex.names
+   MultiIndex.ndim
 
 MultiIndex Modifying and computations
 ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~