熊猫数据框的形状到3d数组

Question

我想将pandas数据框转换为3d数组，但无法获得3d数组的真实形状：

df = pd.DataFrame(np.random.rand(4,5), columns = list('abcde'))

df['a'][3:]=1

df['a'][:3]=2

a3d = np.array(list(df.groupby('a').apply(pd.DataFrame.as_matrix)))

a3d.shape

(2,)

但是，当我这样设置时，我可以得到形状

df = pd.DataFrame(np.random.rand(4,5), columns = list('abcde'))

df['a'][2:]=1

df['a'][:2]=2

a3d = np.array(list(df.groupby('a').apply(pd.DataFrame.as_matrix)))

a3d.shape

(2,2,5)

代码有问题吗？谢谢！