Guest User

Untitled

a guest
Nov 21st, 2017
86
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.85 KB | None | 0 0
  1. tdata = {('A', 50): [1, 2, 3, 4],
  2. ('A', 55): [5, 6, 7, 8],
  3. ('B', 10): [10, 20, 30, 40],
  4. ('B', 20): [50, 60, 70, 80],
  5. ('B', 50): [2, 4, 6, 8],
  6. ('B', 55): [10, 12, 14, 16]}
  7. tdf = pd.DataFrame(tdata, index=range(0,4))
  8.  
  9. A B
  10. 50 55 10 20 50 55
  11. 0 1 5 10 50 2 10
  12. 1 2 6 20 60 4 12
  13. 2 3 7 30 70 6 14
  14. 3 4 8 40 80 8 16
  15.  
  16. print (tdf.drop([('B',10), ('B',20)], axis=1))
  17. A B
  18. 50 55 50 55
  19. 0 1 5 2 10
  20. 1 2 6 4 12
  21. 2 3 7 6 14
  22. 3 4 8 8 16
  23.  
  24. mask = tdf.columns.get_level_values(1) >= 50
  25. print (mask)
  26. [ True True False False True True]
  27.  
  28. print (tdf.loc[:, mask])
  29. A B
  30. 50 55 50 55
  31. 0 1 5 2 10
  32. 1 2 6 4 12
  33. 2 3 7 6 14
  34. 3 4 8 8 16
  35.  
  36. print (tdf.drop([50,55], axis=1, level=1))
  37. B
  38. 10 20
  39. 0 10 50
  40. 1 20 60
  41. 2 30 70
  42. 3 40 80
Add Comment
Please, Sign In to add comment