[MINOR][PYTHON][DOCS] Correct the examples of toPandas and toArrow

zhengruifeng · dongjoon-hyun · commit 56fa1e906583 · 2025-09-16T12:19:08.000-07:00
### What changes were proposed in this pull request? Correct the examples of `toPandas` and `toArrow` ### Why are the changes needed? they are not copy-pastable, and the example of `toArrow` is not correct ### Does this PR introduce _any_ user-facing change? doc-change ### How was this patch tested? manually check, the doc-tests are skipped due to pandas/pyarrow dependency ### Was this patch authored or co-authored using generative AI tooling? no Closes #52353 from zhengruifeng/fix_example_toxxx. Authored-by: Ruifeng Zheng <ruifengz@apache.org> Signed-off-by: Dongjoon Hyun <dongjoon@apache.org>
diff --git a/python/pyspark/sql/dataframe.py b/python/pyspark/sql/dataframe.py
@@ -6510,13 +6510,14 @@ def toArrow(self) -> "pa.Table":
 
         Examples
         --------
+        >>> df = spark.createDataFrame([(2, "Alice"), (5, "Bob")], schema=["age", "name"])
         >>> df.toArrow()  # doctest: +SKIP
         pyarrow.Table
         age: int64
         name: string
         ----
-        age: [[2,5]]
-        name: [["Alice","Bob"]]
+        age: [[2],[5]]
+        name: [["Alice"],["Bob"]]
         """
         ...
 
@@ -6540,6 +6541,7 @@ def toPandas(self) -> "PandasDataFrameLike":
 
         Examples
         --------
+        >>> df = spark.createDataFrame([(2, "Alice"), (5, "Bob")], schema=["age", "name"])
         >>> df.toPandas()  # doctest: +SKIP
            age   name
         0    2  Alice
