python - 切换列中的连续值，直到条件发生变化

Question

我正在尝试比较两个值(这可以是两列甚至是两个随机值)，所述比较的结果将用于填充 pandas 列中的值。比较本身不是问题，可以使用 np.where 完成。但我的目标是有一个条件开关。例如，如果比较结果为真，则在比较再次变为真之前的所有值都将是 X。如果条件再次变为真，则要填充的值将切换为 Y，如果另一个为真，则随后填充 X。 ...直到完成所有比较。
类似这样:Repeat the value in column until a change occurs

创建数据框的最小工作示例:

import pandas as pd
import numpy as np

df = pd.DataFrame(np.random.uniform(0, .01, size=(100, 1)),
                  columns=list('A'))
df['condition'] = np.where(df['A'] > np.random.uniform(0.0, 0.5),
                           'Switch', 'DoNotSwitch')

print df

           A    condition
0   0.004392  DoNotSwitch
1   0.002493  DoNotSwitch
2   0.007735  DoNotSwitch
3   0.000523  Switch
4   0.009902  DoNotSwitch
5   0.001712  DoNotSwitch
6   0.006146  DoNotSwitch
7   0.005325  Switch
8   0.003555  DoNotSwitch
9   0.003444  DoNotSwitch
10  0.005225  DoNotSwitch
11  0.000619  DoNotSwitch

我想要的是:

           A    condition    Result
0   0.004392  DoNotSwitch    T
1   0.002493  DoNotSwitch    T
2   0.007735  DoNotSwitch    T
3   0.000523  Switch         F
4   0.009902  DoNotSwitch    F
5   0.001712  DoNotSwitch    F
6   0.006146  DoNotSwitch    F
7   0.005325  Switch         T
8   0.003555  DoNotSwitch    T
9   0.003444  DoNotSwitch    T
10  0.005225  DoNotSwitch    T
11  0.000619  DoNotSwitch    T

有没有办法在一个步骤中压缩 np.where 和 Result 列的生成，因为 condition 列仅指示应该在哪里进行切换并且不需要切换本身？

Answer 1

国际工业联合会，

s = df['condition'].eq('Switch')

df['result'] = np.where(s.cumsum() % 2 == 0,
                        'T', 'F')

注意这里的s正是你的条件:

df['A'] > np.random.uniform(0.0, 0.5)

所以你可以像这样绕过df['condition']的构造:

s = df['A'] > np.random.uniform(0.0, 0.5)
df['result'] = np.where(s.cumsum() % 2 == 0, 'T', 'F')

输出:

           A    condition result
0   0.004392  DoNotSwitch      T
1   0.002493  DoNotSwitch      T
2   0.007735  DoNotSwitch      T
3   0.000523       Switch      F
4   0.009902  DoNotSwitch      F
5   0.001712  DoNotSwitch      F
6   0.006146  DoNotSwitch      F
7   0.005325       Switch      T
8   0.003555  DoNotSwitch      T
9   0.003444  DoNotSwitch      T
10  0.005225  DoNotSwitch      T
11  0.000619  DoNotSwitch      T