Как преобразовать сгруппированную серию pandas в массив numpy

Question

df ниже содержит целые числа, сгруппированные по времени. Я пытаюсь преобразовать их в массив numpy.

import pandas as pd
import numpy as np

df = pd.DataFrame({
    'Time' : [1,1,1,2,2,2,3,3,3],        
    'A' : [3, 4, 5, 2, 5, 6, 1, 6, 7], 
    'B' : [2, 4, 5, 2, 5, 5, 2, 6, 5],                           
})

GA = np.array(df.groupby(['Time'])['A'].apply(np.array))

Intended Out:

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

Answer 1

используйте для перечисления до преобразования

np.array(df.groupby('Time')['A'].apply(list).tolist())

Out:

array([[3, 4, 5],
   [2, 5, 6],
   [1, 6, 7]])