How can I create a dataset from a dataframe's subset in pandas?

Question

I have a file like this (0 = not banned; 1 = banned):

id  attribute  lvl  banned

 1    fire     50      0
 2    fire     54      0
 3    water    13      1