Skip to content

Commit 6626a7a

Browse files
author
y-p
committed
BUG: groupby.apply assumed list of np.arrays had elements of equal len GH1735
1 parent f1128dd commit 6626a7a

File tree

1 file changed

+19
-9
lines changed

1 file changed

+19
-9
lines changed

pandas/core/groupby.py

Lines changed: 19 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -1845,18 +1845,28 @@ def _wrap_applied_output(self, keys, values, not_indexed_same=False):
18451845
return self._concat_objects(keys, values,
18461846
not_indexed_same=not_indexed_same)
18471847

1848-
if self.axis == 0:
1849-
stacked_values = np.vstack([np.asarray(x)
1850-
for x in values])
1851-
columns = values[0].index
1852-
index = key_index
1853-
else:
1854-
stacked_values = np.vstack([np.asarray(x)
1848+
try:
1849+
if self.axis == 0:
1850+
1851+
stacked_values = np.vstack([np.asarray(x)
1852+
for x in values])
1853+
columns = values[0].index
1854+
index = key_index
1855+
else:
1856+
stacked_values = np.vstack([np.asarray(x)
18551857
for x in values]).T
1856-
index = values[0].index
1857-
columns = key_index
1858+
1859+
index = values[0].index
1860+
columns = key_index
1861+
1862+
except ValueError:
1863+
#GH1738,, values is list of arrays of unequal lengths
1864+
# fall through to the outer else caluse
1865+
return Series(values, index=key_index)
1866+
18581867
return DataFrame(stacked_values, index=index,
18591868
columns=columns)
1869+
18601870
else:
18611871
return Series(values, index=key_index)
18621872
else:

0 commit comments

Comments
 (0)