Python:如何用零填充？

Question

假设我们有如下数据框:

df = pd.DataFrame({ 'Col1' : ['a', 'a', 'a', 'a', 'b', 'b', 'c', 'c'],
        'col2' : ['0.5', '0.78', '0.78', '0.4', '2', '9', '2', '7',]
        })

我计算了 col1 中所有唯一值的行数。就像 a 有 4 行，b 和 c 各有 2 行，方法是:

df.groupby(['Col1']).size()

我得到的输出是

Col1
a    4
b    2
c    2
dtype: int64

完成此操作后，我想检查 a、b、c 中哪一个具有最大行数(在本例中，a 具有最大行数)并填充其他(b 和 c) 以及最大值和它们所具有的行之间的差值，带零(b 和 c 各有 2 行，因为 4 是最大行数，所以我想用 2 个零填充 b 和 c。必须在末尾添加零。

我想用零填充它，因为我想在所有变量(a、b、c)上应用一个固定大小的窗口来绘制图形。

Answer 1

您可以通过 GroupBy.cumcount 创建计数器, 创建 MultiIndex 和 DataFrame.reindex通过 MultiIndex.from_product 创建的所有组合:

df1 = df.set_index(['Col1', df.groupby('Col1').cumcount()])

mux = pd.MultiIndex.from_product(df1.index.levels, names=df1.index.names)
df2 = df1.reindex(mux, fill_value=0).reset_index(level=1, drop=True).reset_index()
print (df2)
   Col1  col2
0     a   0.5
1     a  0.78
2     a  0.78
3     a   0.4
4     b     2
5     b     9
6     b     0
7     b     0
8     c     2
9     c     7
10    c     0
11    c     0