CODEWITHSUNDEEP
pythonpandas
Dance Party2
Dance Party2

Reputation: 7536

Pandas Get Sequence of 1s and 0s Given Strings

Given the following:

import pandas as pd
df = pd.DataFrame({'a':['K','1','1,2,3']})
df
       a
0      K
1      1
2  1,2,3

I would like to convert the values in column a to a corresponding sequence of 1s and 0s given this map:

K 1 2 3 4 5
1 1 1 1 1 1

If a value is present, a 1 is put in place of a 0. If the value is not present, the place is held by a 0. If no value is present, the sequence would be a string of 6 0s.

So "K" would be: 100000

And "1,2,3" would be: 011100

Desired result:

       a       b
0      K  100000
1      1  010000
2  1,2,3  011100

I tried pd.get_dummies(df['a']), but it that only works where there is only 1 character in the cell.

Thanks in advance!

Upvotes: 1

Views: 114

Answers (2)

BENY
BENY

Reputation: 323326

Let us try get_dummies then reindex

s = df.a.str.get_dummies(',').reindex(columns=['K','1','2','3','4','5'],fill_value=0).astype(str).agg(''.join,1)
0    100000
1    010000
2    011100
dtype: object
df['b'] = s

Upvotes: 1

Andrej Kesely
Andrej Kesely

Reputation: 195543

I hope I understood your question well, but you can use .apply() with custom map:

m = {'K': 1 << 0,
     '1': 1 << 1,
     '2': 1 << 2, 
     '3': 1 << 3,
     '4': 1 << 4,
     '5': 1 << 5}
    
df['b'] = df.a.apply(
        lambda x: '{:06b}'.format(sum(m[v] for v in x.split(',')))[::-1]
    )
print(df)

Prints:

       a       b
0      K  100000
1      1  010000
2  1,2,3  011100

Upvotes: 1

Related Questions