如何计算一列重复不变的符号？

Question

我有个专栏。我怎么能造一个新的列来计数重复的正负号呢？

col1
-5
-3
-7
4
5
-0.5
6
8
9

col1        count_sign
-5              3
-3              3
-7              3
4               2
5               2
-0.5            1
6               3
8               3
9               3

前3行是3，因为我们在前3行有3个负号，然后有2个正号和.

Answer 1

# identify the change of signs among rows,
# making count as NaN, where sign is same, else 1
df['count']=np.where(np.sign(df['col1']).diff().eq(0),
        np.nan,
        1) 
# cumsum to group the rows
df['count']=df['count'].cumsum().ffill()

# groupby to take count of each group of rows and return groupsize using transform
df['count']=df.groupby('count')['col1'].transform('size')
df

col1    count
0   -5.0    3
1   -3.0    3
2   -7.0    3
3   4.0     2
4   5.0     2
5   -0.5    1
6   6.0     3
7   8.0     3
8   9.0     3

将符号添加到计数值

df['count']=np.where(np.sign(df['col1']).diff().eq(0),
        np.nan,
        1) 
df['count']=df['count'].cumsum().ffill()
df['count']=df.groupby('count')['col1'].transform('size')*np.sign(df['col1'])
df

    col1    count
0   -5.0    -3.0
1   -3.0    -3.0
2   -7.0    -3.0
3    4.0     2.0
4    5.0     2.0
5   -0.5    -1.0
6    6.0     3.0
7    8.0     3.0
8    9.0     3.0