Guest User

Untitled

a guest
Jul 17th, 2018
102
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.58 KB | None | 0 0
  1. # Import libraries
  2. import pandas as pd
  3. import numpy as np
  4.  
  5. # Create sample dataframe
  6. df = pd.DataFrame({'a': ['A1', 'A1', 'A2', 'A3', 'A4', 'A3'],
  7. 'value': [1,2,3,4,5,6]})
  8.  
  9. # Calculate count, mean
  10. temp1 = df.groupby(['a']).count().reset_index().rename(columns={'value':'count'})
  11. temp2 = df.groupby(['a'])['value'].mean().reset_index().rename(columns={'value':'mean'})
  12.  
  13. # Add columns to existing dataframe
  14. df.merge(temp1, on='a', how='inner').merge(temp2, on='a', how='inner')
  15.  
  16. # Add columns to a new dataframe
  17. df2 = temp1.merge(temp2, on='a', how='inner')
  18. df2
Add Comment
Please, Sign In to add comment