小编Yan*_*nis的帖子

我有一个Dataframe,我希望使用其中一列作为第三维转换为多维数组.
举个例子:

df = pd.DataFrame({
'id': [1, 2, 2, 3, 3, 3],
'date': np.random.randint(1, 6, 6),
'value1': [11, 12, 13, 14, 15, 16],
'value2': [21, 22, 23, 24, 25, 26]
 })

我想将它转换为具有尺寸(id,日期,值)的3D数组,如下所示:

问题是'id的出现次数不同,所以我不能使用np.reshape().

对于这个简化的例子,我能够使用:

ra = np.full((3, 3, 3), np.nan)

for i, value in enumerate(df['id'].unique()):
    rows = df.loc[df['id'] == value].shape[0]
    ra[i, :rows, :] = df.loc[df['id'] == value, 'date':'value2']

为了产生所需的结果:

但原始DataFrame包含数百万行.

是否有一种矢量化方式来帮助同样的结果？