Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- tdata = {('A', 50): [1, 2, 3, 4],
- ('A', 55): [5, 6, 7, 8],
- ('B', 10): [10, 20, 30, 40],
- ('B', 20): [50, 60, 70, 80],
- ('B', 50): [2, 4, 6, 8],
- ('B', 55): [10, 12, 14, 16]}
- tdf = pd.DataFrame(tdata, index=range(0,4))
- A B
- 50 55 10 20 50 55
- 0 1 5 10 50 2 10
- 1 2 6 20 60 4 12
- 2 3 7 30 70 6 14
- 3 4 8 40 80 8 16
- print (tdf.drop([('B',10), ('B',20)], axis=1))
- A B
- 50 55 50 55
- 0 1 5 2 10
- 1 2 6 4 12
- 2 3 7 6 14
- 3 4 8 8 16
- mask = tdf.columns.get_level_values(1) >= 50
- print (mask)
- [ True True False False True True]
- print (tdf.loc[:, mask])
- A B
- 50 55 50 55
- 0 1 5 2 10
- 1 2 6 4 12
- 2 3 7 6 14
- 3 4 8 8 16
- print (tdf.drop([50,55], axis=1, level=1))
- B
- 10 20
- 0 10 50
- 1 20 60
- 2 30 70
- 3 40 80
Add Comment
Please, Sign In to add comment