Guest User

Untitled

a guest
Jun 23rd, 2018
95
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.46 KB | None | 0 0
  1. import pandas as pd
  2.  
  3. numbers = [9, 23, 33, 91, 13]
  4. players = ["Ron Harper", "Michael Jordan", "Scottie Pippen", "Dennis Rodman", "Luc Longley"]
  5. weights = [185, 195, 210, 210, 265]
  6. df = pd.DataFrame()
  7. df["number"] = numbers
  8. df["player"] = players
  9. df["weight"] = weights
  10.  
  11. def get_weight_category(wt):
  12. if wt < 200:
  13. return "Light"
  14. elif 200 <= wt < 250:
  15. return "Medium"
  16. else:
  17. return "Heavy"
  18.  
  19. df["weight_category"] = df["weight"].map(get_weight_category)
  20. df
Add Comment
Please, Sign In to add comment