python - 如何将 numpy 数组分成 pandas 中的单独列

Question

我有一个看起来像的数据框

  ID_0 ID_1  ID_2
0    a    b  0.05
1    a    b  0.10
2    a    b  0.19
3    a    c  0.25
4    a    c  0.40
5    a    c  0.65
6    a    c  0.71
7    d    c  0.95
8    d    c  1.00

我想分组并为每个组制作 ID_2 列的标准化直方图。所以我愿意

df.groupby(['ID_0', 'ID_1']).apply(lambda x: np.histogram(x['ID_2'], range = (0,1), density=True)[0]).reset_index(name='ID_2')

但是我真正想要的是 numpy 数组的 11 个元素位于数据帧的单独列中。

我该怎么做？

Answer 1

您可以从每个 numpy 数组构造一个系列对象，并且元素将作为列进行广播:

import pandas as pd
import numpy as np
df.groupby(['ID_0', 'ID_1']).apply(lambda x: pd.Series(np.histogram(x['ID_2'], range = (0,1), density=True)[0])).reset_index()