大熊貓據幀值與numpy的陣列

Question

例如我有這樣一個數據幀：

import pandas as pd 

df = pd.DataFrame([[1, 2.], [3, 4.]], columns=['a', 'b']) 

print df 

    a b 
0 1 2.0 
1 3 4.0 

我希望得到一個數據幀如下：

a  b 
0 [1,3] [2,4] 

Answer 1

一種方法 -

df_out = pd.DataFrame([df.values.T.astype(int).tolist()], columns=df.columns) 

收回 -

N = len(df_out.columns) 
arr_back = np.concatenate(np.concatenate(df_out.values)).reshape(N,-1).T 
df_back = pd.DataFrame(arr_back, columns=df_out.columns) 

採樣運行 -

In [164]: df 
Out[164]: 
    a b 
0 1 2.0 
1 3 4.0 
2 5 6.0 

In [165]: df_out 
Out[165]: 
      a   b 
0 [1, 3, 5] [2, 4, 6] 

In [166]: df_back 
Out[166]: 
    a b 
0 1 2 
1 3 4 
2 5 6