dataframe填充列带有先前的值，直到条件

Question

我有一个看起来像这样的数据框架：

步骤	文本	参数
15	打印	1
16	控件	2
17	打印输出	3
18	print2	1
19	nan	2
20	nan	3 22 nan 3
21	nan	4
22	nan	1
23	nan 2 nan	2
24	nan	1

，我希望我的dataframe看起来像此：

步骤	文本	参数
15	打印	1
15	打印	2
15	打印	3
16	控制	1
16	控制	2
17	控制	3
17	控制	4
18	打印输出	1
18	打印输出	2
19	print 2	1

因此，基本上我在参数列中“ 1”时，我需要步骤和文本的下一个值。 有什么想法吗？:)

Answer 1

您可以在自定义组上使用重复：

# ensure NaN
df['Text'] = df['Text'].replace('Nan', pd.NA)

# get the number of rows per group starting with 1
n = df.groupby(df['Parameter'].eq(1).cumsum()).size()

# repeat the index of the non NaN values as many times
idx = df['Text'].dropna().index.repeat(n)

# replace the values ignoring the index
# (using the underlying numpy array)
df[['Step', 'Text']] = df.loc[idx, ['Step', 'Text']].to_numpy()

输出：

  Step      Text  Parameter
0   15     print          1
1   15     print          2
2   15     print          3
3   16   control          1
4   16   control          2
5   16   control          3
6   16   control          4
7   17  printout          1
8   17  printout          2
9   18    print2          1