如何将分组的熊猫系列转换为 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))

预期：

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

Answer 1

转换前使用 to 列出

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

出去：

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