Конвертировать pandas массив данных списков в массив numpy

Question

У меня есть следующий фрейм данных:

import pandas as pd
import numpy as np

df = pd.DataFrame([{'a': [1,3,2]},{'a': [7,6,5]},{'a': [9,8,8]}])
df

df['a'].to_numpy()
df['a'].to_numpy()

=> array([list([1, 3, 2]), list([7, 6, 5]), list([9, 8, 8])], dtype=object)

Как получить массив numpy формы (3,3) без записи для l oop?

Answer 1

Сначала создайте вложенные списки, а затем преобразуйте в массив, необходимы только все списки одинаковой длины:

arr = np.array(df.a.tolist())
print (arr)
[[1 3 2]
 [7 6 5]
 [9 8 8]]

Answer 2

Если всегда иметь одинаковую длину

pd.DataFrame(df.a.tolist()).values
array([[1, 3, 2],
       [7, 6, 5],
       [9, 8, 8]])