create a dictionary of array for each cell within dataframe

Question

i have a dataframe with two columns of array.

Column 1
0        [254.0, 254.0, 254.0, 254.0, 254.0]
1        [254.0, 254.0, 254.0, 254.0, 254.0]
2        [254.0, 254.0, 254.0, 254.0, 254.0]
3        [254.0, 254.0, 254.0, 254.0, 254.0]
4        [254.0, 254.0, 254.0, 254.0, 254.0]

Column 2
0      [3.0, 4.0, 7.0, 1.0, 5.0]  
1      [3.0, 4.0, 7.0, 1.0, 5.0]  
2      [3.0, 4.0, 7.0, 1.0, 5.0]  
3      [3.0, 4.0, 7.0, 1.0, 5.0]  
4      [3.0, 4.0, 7.0, 1.0, 5.0]

I am trying to create a third column where each cell looks like a dictionary of arrays. It would look like the following:

Column 3
0        {0:[254.0,3], 1:[254.0,4], 2:[254.0,7], 3:[254.0,1], 4:[254.0,5]}
1        {0:[254.0,3], 1:[254.0,4], 2:[254.0,7], 3:[254.0,1], 4:[254.0,5]}
2        {0:[254.0,3], 1:[254.0,4], 2:[254.0,7], 3:[254.0,1], 4:[254.0,5]}
3        {0:[254.0,3], 1:[254.0,4], 2:[254.0,7], 3:[254.0,1], 4:[254.0,5]}
4        {0:[254.0,3], 1:[254.0,4], 2:[254.0,7], 3:[254.0,1], 4:[254.0,5]}

how do i do this ?

Quang Hoang · Accepted Answer · 2022-08-07 05:10:33Z

2

How about:

df['Column 3'] = df.apply(lambda x: dict(enumerate(zip(x['Column 1'], x['Column 2']))), axis=1)

answered Aug 7, 2022 at 5:10

Quang Hoang

151k11 gold badges64 silver badges86 bronze badges

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

Comments

Michael Hodel · Accepted Answer · 2022-08-07 05:08:53Z

import pandas as pd

c1 = [[254.0, 254.0, 254.0, 254.0, 254.0],
[254.0, 254.0, 254.0, 254.0, 254.0],
[254.0, 254.0, 254.0, 254.0, 254.0],
[254.0, 254.0, 254.0, 254.0, 254.0],
[254.0, 254.0, 254.0, 254.0, 254.0]]

c2 = [[3.0, 4.0, 7.0, 1.0, 5.0],
[3.0, 4.0, 7.0, 1.0, 5.0], 
[3.0, 4.0, 7.0, 1.0, 5.0],
[3.0, 4.0, 7.0, 1.0, 5.0], 
[3.0, 4.0, 7.0, 1.0, 5.0]]

df = pd.DataFrame({'Column 1': c1, 'Column 2': c2})
df['Column 3'] = df.apply(lambda r: {i: [v1, v2] for i, (v1, v2) in enumerate(zip(r['Column 1'], r['Column 2']))}, axis=1)
df

prints

index	Column 1	Column 2	Column 3
0	254,254,254,254,254	3,4,7,1,5	{0: [254.0, 3.0], 1: [254.0, 4.0], 2: [254.0, 7.0], 3: [254.0, 1.0], 4: [254.0, 5.0]}
1	254,254,254,254,254	3,4,7,1,5	{0: [254.0, 3.0], 1: [254.0, 4.0], 2: [254.0, 7.0], 3: [254.0, 1.0], 4: [254.0, 5.0]}
2	254,254,254,254,254	3,4,7,1,5	{0: [254.0, 3.0], 1: [254.0, 4.0], 2: [254.0, 7.0], 3: [254.0, 1.0], 4: [254.0, 5.0]}
3	254,254,254,254,254	3,4,7,1,5	{0: [254.0, 3.0], 1: [254.0, 4.0], 2: [254.0, 7.0], 3: [254.0, 1.0], 4: [254.0, 5.0]}
4	254,254,254,254,254	3,4,7,1,5	{0: [254.0, 3.0], 1: [254.0, 4.0], 2: [254.0, 7.0], 3: [254.0, 1.0], 4: [254.0, 5.0]}

Collectives™ on Stack Overflow

create a dictionary of array for each cell within dataframe

2 Answers 2

Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related