Python: Multidimensional dictionary to array

Question

I would like to convert a dictionary of the form:

1: [' Ma','Ant','Man','io'] 2: [' Sc','Alb','Man'] 3: [' Sc','Alb','Sch','bre']

to a matrix where all the possible values are the columns and keys are indices. Each cell of the matrix should contain 1 if the corresponding value ( column header) is present in the key and 0 otherwise.

' Ma' 'Ant' 'Man' 'io' ' Sc','Alb','Sch','bre' 1: 1 1 1 1 0 0 0 0 2: 0 0 1 0 1 1 0 0 3: 0 0 0 0 1 1 1 1

I don't know from where to start and how to use pandas and Dataframe to get this done.

root · Accepted Answer · 2016-05-24 16:10:32Z

3

You can use pd.DataFrame.from_dict to load the dictionary, then use pd.get_dummies to get the 0/1 values:

d = {1: ['Ma','Ant','Man','io'], 2: ['Sc','Alb','Man'], 3: ['Sc','Alb','Sch','bre']}
df = pd.DataFrame.from_dict(d, orient='index')
df = pd.get_dummies(df, prefix='', prefix_sep='').astype(int)

The resulting output:

   Ma  Sc  Alb  Ant  Man  Sch  bre  io
1   1   0    0    1    1    0    0   1
2   0   1    1    0    1    0    0   0
3   0   1    1    0    0    1    1   0

answered May 24, 2016 at 16:10

root

34.1k6 gold badges77 silver badges89 bronze badges

Sign up to request clarification or add additional context in comments.

1 Comment

user2804064 Over a year ago

Perfect. Thank you very much!

Collectives™ on Stack Overflow

Python: Multidimensional dictionary to array

1 Answer 1

1 Comment

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

1 Comment

Your Answer

Sign up or log in

Post as a guest

Related