Pandas 中带有索引移位的条件置换

Question

在数据框中有以下列：

0
0
0
0
0
5

我想检查大于阈值的值。如果找到，则设置为零并向上移动差异值阈值，在新位置上设置阈值。假设阈值=3，那么结果列必须是：

0
0
0
3
0
0

有快速转型的想法吗？

Answer 1

对于这个数据框：

df
出去：
    一个
0 0
1 0
2 0
3 0
4 0
5 5
6 0
7 0
8 0
9 0
10 6
11 0
12 0

threshold = 3
above_threshold = df['A'] > threshold
df.loc[df[above_threshold].index - (df.loc[above_threshold, 'A'] - 3).values, 'A'] = 3
df.loc[above_threshold, 'A'] = 0

df
出去：
    一个
0 0
1 0
2 0
3 3
4 0
5 0
6 0
7 3
8 0
9 0
10 0
11 0
12 0