Skip to content

Commit 6ab69e0

Browse files
author
Mark Graham
authored
DOC: add two new examples to the transform docs (pandas-dev#37912)
1 parent f4fab6d commit 6ab69e0

File tree

1 file changed

+53
-0
lines changed

1 file changed

+53
-0
lines changed

pandas/core/shared_docs.py

+53
Original file line numberDiff line numberDiff line change
@@ -324,4 +324,57 @@
324324
0 0.000000 1.000000
325325
1 1.000000 2.718282
326326
2 1.414214 7.389056
327+
328+
You can call transform on a GroupBy object:
329+
330+
>>> df = pd.DataFrame({{
331+
... "Date": [
332+
... "2015-05-08", "2015-05-07", "2015-05-06", "2015-05-05",
333+
... "2015-05-08", "2015-05-07", "2015-05-06", "2015-05-05"],
334+
... "Data": [5, 8, 6, 1, 50, 100, 60, 120],
335+
... }})
336+
>>> df
337+
Date Data
338+
0 2015-05-08 5
339+
1 2015-05-07 8
340+
2 2015-05-06 6
341+
3 2015-05-05 1
342+
4 2015-05-08 50
343+
5 2015-05-07 100
344+
6 2015-05-06 60
345+
7 2015-05-05 120
346+
>>> df.groupby('Date')['Data'].transform('sum')
347+
0 55
348+
1 108
349+
2 66
350+
3 121
351+
4 55
352+
5 108
353+
6 66
354+
7 121
355+
Name: Data, dtype: int64
356+
357+
>>> df = pd.DataFrame({{
358+
... "c": [1, 1, 1, 2, 2, 2, 2],
359+
... "type": ["m", "n", "o", "m", "m", "n", "n"]
360+
... }})
361+
>>> df
362+
c type
363+
0 1 m
364+
1 1 n
365+
2 1 o
366+
3 2 m
367+
4 2 m
368+
5 2 n
369+
6 2 n
370+
>>> df['size'] = df.groupby('c')['type'].transform(len)
371+
>>> df
372+
c type size
373+
0 1 m 3
374+
1 1 n 3
375+
2 1 o 3
376+
3 2 m 4
377+
4 2 m 4
378+
5 2 n 4
379+
6 2 n 4
327380
"""

0 commit comments

Comments
 (0)