Skip to content

Commit fa8bba2

Browse files
committed
Prevent error when creating ps.Series from ps.Series
1 parent 063144d commit fa8bba2

File tree

2 files changed

+10
-1
lines changed

2 files changed

+10
-1
lines changed

python/pyspark/pandas/series.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -398,12 +398,15 @@ def __init__( # type: ignore[no-untyped-def]
398398

399399
self._anchor: DataFrame # type: ignore[annotation-unchecked]
400400
self._col_label: Label # type: ignore[annotation-unchecked]
401-
if isinstance(data, DataFrame):
401+
if isinstance(data, (DataFrame, Series)):
402402
assert dtype is None
403403
assert name is None
404404
assert not copy
405405
assert not fastpath
406406

407+
if isinstance(data, Series):
408+
data = DataFrame(data)
409+
407410
self._anchor = data
408411
self._col_label = index
409412
else:

python/pyspark/pandas/tests/series/test_series.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -103,6 +103,12 @@ def test_empty_series(self):
103103

104104
self.assertTrue(pser_a.empty)
105105

106+
def test_series_from_series(self):
107+
pser = ps.Series([1, 2, 3, 4, 5, 6, 7], name="x")
108+
109+
self.assert_eq(ps.Series(pser), pser)
110+
self.assert_eq(ps.Series(pser, name="y"), pser.rename("y"))
111+
106112
def test_all_null_series(self):
107113
pser_a = pd.Series([None, None, None], dtype="float64")
108114
pser_b = pd.Series([None, None, None], dtype="str")

0 commit comments

Comments
 (0)