How to convert a grouped pandas series to a numpy array

Question

The df below contains integers grouped by time. I’m trying to convert these to a numpy array.

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]]

Naga kiran · Accepted Answer · 2020-03-12 02:31:15Z

2

use to to list before converting

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

Out:

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

answered Mar 12, 2020 at 2:31

Naga kiran

4,6071 gold badge21 silver badges32 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

How to convert a grouped pandas series to a numpy array

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related