[
https://issues.apache.org/jira/browse/SPARK-36907?page=com.atlassian.jira.plugin.system.issuetabpanels:comment-tabpanel&focusedCommentId=17423054#comment-17423054
]
Apache Spark commented on SPARK-36907:
--------------------------------------
User 'ueshin' has created a pull request for this issue:
https://github.com/apache/spark/pull/34160
> pandas API on Spark: DataFrameGroupBy.apply raises an exception when it
> returns Series.
> ---------------------------------------------------------------------------------------
>
> Key: SPARK-36907
> URL: https://issues.apache.org/jira/browse/SPARK-36907
> Project: Spark
> Issue Type: Bug
> Components: PySpark
> Affects Versions: 3.2.0, 3.3.0
> Reporter: Takuya Ueshin
> Priority: Major
>
> {{DataFrameGroupBy.apply}} without shortcut could raise an exception when it
> returns {{Series}}.
> {code:python}
> >>> ps.options.compute.shortcut_limit = 3
> >>> psdf = ps.DataFrame(
> ... {"a": [1, 2, 3, 4, 5, 6], "b": [1, 1, 2, 3, 5, 8], "c": [1, 4, 9, 16,
> 25, 36]},
> ... columns=["a", "b", "c"],
> ... )
> >>> psdf.groupby("b").apply(lambda x: x["a"])
> org.apache.spark.api.python.PythonException: Traceback (most recent call
> last):
> ...
> ValueError: Length mismatch: Expected axis has 2 elements, new values have 3
> elements
> {code}
--
This message was sent by Atlassian Jira
(v8.3.4#803005)
---------------------------------------------------------------------
To unsubscribe, e-mail: issues-unsubscr...@spark.apache.org
For additional commands, e-mail: issues-h...@spark.apache.org
