How to convert list to a list of numpy arrays?

I have the following list [1 2 3 4 5 6 7 8]. How can I convert this to a list of [[1 2 3] 4 5 [6 7 8]].

[[1 2 3] 4 5 [6 7 8]]

Better yet, I have 1 2 3 4 5 6 7 8 as a row in my dataframe and I'd like to convert my dataframe to instead have 4 columns where the first entry in this row is [1 2 3], the second one is 4, and the third and fourth entries are 5 and [6 7 8]. Of course I'd like to do that for every row in my dataframe too, not just this row.

1 2 3 4 5 6 7 8

[1 2 3]

4

5

[6 7 8]

Anyone know a way to do this?

[[1 2 3] 4 5 [6 7 8]]

1 Answer
1

IIUC

df = pd.DataFrame({k: k for k in range(1,9)}, index=range(5))

1 2 3 4 5 6 7 8
0 1 2 3 4 5 6 7 8
1 1 2 3 4 5 6 7 8
2 1 2 3 4 5 6 7 8
3 1 2 3 4 5 6 7 8
4 1 2 3 4 5 6 7 8

>>> pd.DataFrame({'0': df[[1,2,3]].agg(np.array, 1),
'1': df[4],
'2': df[5],
'3': df[[6,7,8]].agg(np.array, 1)})

0 1 2 3
0 [1, 2, 3] 4 5 [6, 7, 8]
1 [1, 2, 3] 4 5 [6, 7, 8]
2 [1, 2, 3] 4 5 [6, 7, 8]
3 [1, 2, 3] 4 5 [6, 7, 8]
4 [1, 2, 3] 4 5 [6, 7, 8]


print(df.head(5).to_dict())

.agg

By clicking "Post Your Answer", you acknowledge that you have read our updated terms of service, privacy policy and cookie policy, and that your continued use of the website is subject to these policies.