python - Pandas :在数据框中的第二行之后插入一个空行

到目前为止，我有这段代码，每隔一行（from this question）添加零行：

import pandas as pd
import numpy as np

def Add_Zeros(df):

    zeros = np.where(np.empty_like(df.values), 0, 0)
    data = np.hstack([df.values, zeros]).reshape(-1, df.shape[1])
    df_ordered = pd.DataFrame(data, columns=df.columns)

    return df_ordered

结果为以下数据帧：

    A   B
0   a   a
1   0   0
2   b   b
3   0   0
4   c   c
5   0   0
6   d   d

但是我需要它在每第二行添加零行，如下所示：

    A   B
0   a   a
1   b   b
2   0   0
3   c   c
4   d   d
5   0   0

我尝试更改代码，但每次都会收到一条错误消息，指出零和df大小不匹配。

我还应该指出，我的行和列比这里写的要多得多。

我怎样才能做到这一点？

最佳答案

选项1
使用groupby

s = pd.Series(0, df.columns)
f = lambda d: d.append(s, ignore_index=True)

grp = np.arange(len(df)) // 2
df.groupby(grp, group_keys=False).apply(f).reset_index(drop=True)

   A  B
0  a  a
1  b  b
2  0  0
3  c  c
4  d  d
5  0  0

选项2

from itertools import repeat, chain

v = df.values

pd.DataFrame(
    np.row_stack(list(chain(*zip(v[0::2], v[1::2], repeat(z))))),
    columns=df.columns
)

   A  B
0  a  a
1  b  b
2  0  0
3  c  c
4  d  d
5  0  0

关于python - Pandas :在数据框中的第二行之后插入一个空行，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/47148170/