回答此问题可获得 20 贡献值,回答如果被采纳可获得 50 分。
<p>我的目标是使用一个条件来子集一个df,并将这些行附加到df的右侧。例如,当<code>Num2</code>等于<code>1</code>时,我想取下一行并将其附加到<code>df</code>的右侧。下面每</em>行追加<em>,其中我只想在<code>Num2</code>中的<code>1</code>之后追加下一行。我还希望能够附加特定的COL。使用下面的方法,这只能是Num1和Num2</p>
<pre><code>df = pd.DataFrame({
'Num1' : [0,1,2,3,4,4,0,1,2,3,1,1,2,3,4,0],
'Num2' : [0,0,0,0,0,1,3,0,1,2,0,0,0,0,1,4],
'Value' : [0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0]
})
df1 = df.add_suffix('1').join(df.shift(-1).add_suffix('2'))
</code></pre>
<p>预期产出:</p>
<pre><code># grab all rows after a 1 in Num2
ones = df.loc[df["Num2"].shift().isin([1])]
# append these to the right
Num1 Num2 Value Num12 Num22
0 0 0 0
1 1 0 0
2 2 0 0
3 3 0 0
4 4 0 0
5 4 1 0 0 3
6 0 3 0
7 1 0 0
8 2 1 0 3 2
9 3 2 0
10 1 0 0
11 1 0 0
12 2 0 0
13 3 0 0
14 4 1 0 0 4
15 0 4 0
</code></pre>