pandas - How to use MultiLabelBinarizer for Multilabel classification?

Question

I am trying to do multilabel classification. But I am really stuck at data preprocessing. My target data in in a separate file. The target data looks like this

   Id              Tag
0   1             data
1   4               c#
2   4         winforms
3   4  type-conversion
4   4          decimal

I am trying to use MultiLabelBinarizer to preprocess the data. At the end of it, I want it to look something like this -

ID	data	c#	winforms	type-conversion	decimal
1	1	0	0	0	0
4	0	1	1	1	1

Answer 1

You are not doing anything wrong. MultiLabelBinarizer(), as most other sklearn stuff, returns numpy arrays. In this case, the underlying data looks identical to your expected output, sans the ID and Tag names.

Use pd.crosstab instead:

pd.crosstab(df['Id'], df['Tag'])