преобразовать каждую строку таблицы панд в список и добавить в таблицу

Question

Я хотел бы преобразовать каждую строку в список и вставить в качестве нового столбца.Например, я начинаю со следующей таблицы:

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

и хотел бы получить:

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

Вот мой фрагмент кода, который я ожидал достичь именно этого, только это не так't.

import pandas as pd
import numpy as np

df = pd.DataFrame(np.random.randint(low=0, high=10, size=(5, 5)))
df['list'] = df.apply(lambda x: x.tolist(), axis=1)
print(df)

Answer 1

Вы можете использовать:

In [127]: df['test'] = df.values.tolist()

In [128]: df
Out[128]: 
   0  1  2  3  4             test
0  0  8  3  2  5  [0, 8, 3, 2, 5]
1  1  1  2  2  4  [1, 1, 2, 2, 4]
2  0  8  9  6  4  [0, 8, 9, 6, 4]
3  2  7  6  1  9  [2, 7, 6, 1, 9]
4  8  9  1  5  6  [8, 9, 1, 5, 6]