pandas group by and assign a group id then ungroup

by

By using ngroup

df['grpId']=df.groupby(' socialmedia').ngroup().add(1)
df
Out[354]: 
    id  socialmedia  grpId
0    1     facebook      1
1    2     facebook      1
2    3       google      2
3    4       google      2
4    5       google      2
5    6      twitter      4
6    7       google      2
7    8      twitter      4
8    9     snapchat      3
9   10      twitter      4
10  11     facebook      1

Or pd.factorize and 'categroy'

df['grpId']=pd.factorize(df[' socialmedia'])[0]+1

df
Out[358]: 
    id  socialmedia  grpId
0    1     facebook      1
1    2     facebook      1
2    3       google      2
3    4       google      2
4    5       google      2
5    6      twitter      3
6    7       google      2
7    8      twitter      3
8    9     snapchat      4
9   10      twitter      3
10  11     facebook      1

df['grpId']=df[' socialmedia'].astype('category').cat.codes.add(1)
df
Out[356]: 
    id  socialmedia  grpId
0    1     facebook      1
1    2     facebook      1
2    3       google      2
3    4       google      2
4    5       google      2
5    6      twitter      4
6    7       google      2
7    8      twitter      4
8    9     snapchat      3
9   10      twitter      4
10  11     facebook      1

More Related Contents:

  1. Calculate cumulative intraday measures that reset every day in pandas
  2. Get the row(s) which have the max value in groups using groupby
  3. Get statistics for each group (such as count, mean, etc) using pandas GroupBy?
  4. How do I create a new column from the output of pandas groupby().sum()?
  5. Aggregation in Pandas
  6. Pandas groupby with delimiter join
  7. Pandas: filling missing values by mean in each group
  8. Pandas GroupBy.apply method duplicates first group
  9. Keep other columns when doing groupby
  10. pandas GroupBy columns with NaN (missing) values
  11. Count unique values per groups with Pandas [duplicate]
  12. Multiple aggregations of the same column using pandas GroupBy.agg()
  13. Pandas groupby mean – into a dataframe?
  14. Pandas, groupby and count
  15. Pandas GroupBy and select rows with the minimum value in a specific column
  16. Pandas Groupby and Sum Only One Column
  17. Python Pandas Group by date using datetime data
  18. What is the difference between pandas agg and apply function?
  19. why does pandas rolling use single dimension ndarray
  20. Pandas Groupby / List to Multiple Rows
  21. Pandas – dataframe groupby – how to get sum of multiple columns
  22. Group dataframe and get sum AND count?
  23. Pandas groupby with categories with redundant nan
  24. Bar graph from dataframe groupby
  25. Select the max row per group – pandas performance issue
  26. How to summarize on different groupby combinations?
  27. What is the equivalent of SQL “GROUP BY HAVING” on Pandas?
  28. Pandas groupby and aggregation output should include all the original columns (including the ones not aggregated on)
  29. Get the row corresponding to the max in pandas GroupBy [duplicate]
  30. When is it appropriate to use df.value_counts() vs df.groupby(‘…’).count()?

Leave a Comment