[pandas] Convert categorical data to numbers

[ThomasL]

A possibility that might be useful for you:

import pandas as pd

s = pd.Series(['ABC_commercial', 'ABC_Private', 'Test ROM DIV', 'ROM DIV', 'TEST SEC ROM'], dtype="object")
df = pd.DataFrame(s, columns=['sub_division'])

df['ABC'] = (df.sub_division.str.find('ABC_') > -1) * 1
df['ROM'] = (df.sub_division.str.find('ROM') > -1) * 1

print(df)

Output:

     sub_division  ABC  ROM
0  ABC_commercial    1    0
1     ABC_Private    1    0
2    Test ROM DIV    0    1
3         ROM DIV    0    1
4    TEST SEC ROM    0    1