パンダ内でのnpの変形適用

Question

発生例外：データは1次元でなければなりません。

私はこの問題を明確にするためにおもちゃの例を提示します。

import pandas as pd 
import numpy as np 

初期データフレーム：numpyの配列に

df = pd.DataFrame({"A": [[10,15,12,14],[20,30,10,43]], "R":[2,2] ,"C":[2,2]}) 
>>df 

     A     C R 
0 [10, 15, 12, 14] 2 2 
1 [20, 30, 10, 43] 2 2 

変換と再構築：

df['A'] = df['A'].apply(lambda x: np.array(x)) 
df.apply(lambda x: print(x[0],(x[1],x[2])) ,axis=1) 
df['A_reshaped'] = df.apply(lambda x[['A','R','C']]: np.reshape(x[0],(x[1],x[2])),axis=1) 
df 

     A     C R   A_reshaped 
0 [10, 15, 12, 14] 2 2  [[10,15],[12,14]] 
1 [20, 30, 10, 43] 2 2  [[20,30],[10,43]] 

を

誰かが理由を知っていますか？それはパンダの細胞の2次元配列を受け入れていないようだが、それは奇妙だ...

事前にお手伝いを！

Answer 1

applyを使用すると、戻り値はnumpy 2d配列になり、DataFrameに戻すと何らかの理由でPandasが混乱することがあります。

これが動作しているようです、しかし：

df['reshaped'] = pd.Series([a.reshape((c, r)) for (a, c, r) in zip(df.A, df.C, df.R)]) 

>>> df 
        A C R    reshaped 
0 [10, 15, 12, 14] 2 2 [[10, 15], [12, 14]] 
1 [20, 30, 10, 43] 2 2 [[20, 30], [10, 43]]