在列中保持第一个值，而另一列等于值？

Question

我想保持列中的第一个值，而另一列不等于零。对于列B，值在-1,0,1之间交替。对于列C，值等于任何整数。目标是保持列C的第一个值，而列B等于零。目前的数据框如下：

A B C 
1 8 1 9 
2 2 1 1 
3 3 0 7 
4 9 0 8 
5 5 0 9 
6 6 0 1 
7 1 1 9 
8 6 1 10 
9 3 0 4 
10 8 0 8 
11 5 0 9 
12 6 0 10 

产生的数据帧应该如下：

A B C 
1 8 1 9 
2 2 1 1 
3 3 0 7 
4 9 0 7 
5 5 0 7 
6 6 0 7 
7 1 1 9 
8 6 1 10 
9 3 0 4 
10 8 0 4 
11 5 0 4 
12 6 0 4 
13 3 1 9 

Answer 1

您需要先创建NaN S按条件C列，然后通过ffill添加值：

mask = (df['B'].shift().fillna(False)).astype(bool) | (df['B']) 
df['C'] = df.loc[mask, 'C'] 
df['C'] = df['C'].ffill().astype(int) 
print (df) 
    A B C 
1 8 1 9 
2 2 1 1 
3 3 0 7 
4 9 0 7 
5 5 0 7 
6 6 0 7 
7 1 1 9 
8 6 1 10 
9 3 0 4 
10 8 0 4 
11 5 0 4 
12 6 0 4 
13 3 1 9 

或使用where如果所有值的类型都是整数，则添加astype：

mask = (df['B'].shift().fillna(False)).astype(bool) | (df['B']) 
df['C'] = df['C'].where(mask).ffill().astype(int) 
print (df) 
    A B C 
1 8 1 9 
2 2 1 1 
3 3 0 7 
4 9 0 7 
5 5 0 7 
6 6 0 7 
7 1 1 9 
8 6 1 10 
9 3 0 4 
10 8 0 4 
11 5 0 4 
12 6 0 4 
13 3 1 9