如何将分组的熊猫系列转换为numpy数组

时间:2020-03-12 02:27:15

标签: python pandas numpy

下面的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]]

1 个答案:

答案 0 :(得分:2)

用于在转换之前列出

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

出局:

array([[3, 4, 5],
   [2, 5, 6],
   [1, 6, 7]])
相关问题