SNOW-2405552: Reuse row_count from the relaxed query compiler in get_axis_len() (#3878)

Author: sfc-gh-helmeleegy

CHANGELOG.md

@@ -70,6 +70,7 @@
   - `str.endswith`
   - `str.slice`
   - `sort_values`
+- Reuse row count from the relaxed query compiler in `get_axis_len`.
 
 #### Bug Fixes
 

src/snowflake/snowpark/modin/plugin/compiler/snowflake_query_compiler.py

@@ -10692,7 +10692,21 @@ def get_axis_len(
         -------
         Length of the specified axis.
         """
-        return self._modin_frame.num_rows if axis == 0 else len(self.columns)
+        if axis == 0:
+            if (
+                self._relaxed_query_compiler is not None
+                and self._relaxed_query_compiler._modin_frame.ordered_dataframe.row_count
+                is not None
+            ):
+                row_count = (
+                    self._relaxed_query_compiler._modin_frame.ordered_dataframe.row_count
+                )
+                self._modin_frame.ordered_dataframe.row_count = row_count
+                return row_count
+            else:
+                return self._modin_frame.num_rows
+        else:
+            return len(self.columns)
 
     def _nunique_columns(
         self, dropna: bool, include_index: bool = False