pandas 查找非空值的起点和终点

Question

我想找到一列的起点和终点并对其进行标记，如下所示：

价值	旗帜
南	南
南	南
1	开始
2	南
1	南
3	南
2	停止
南	南
1	开始
2	停止

Answer 1

• start当当前值notnull与前一个值相同时发生isnull
• stop当当前值是notnull和下一个值时发生isnull

使用生成这些条件shift并使用分配loc：

start = df.value.notnull() & df.value.shift().isnull()
stop = df.value.notnull() & df.value.shift(-1).isnull()

df.loc[start, 'flag'] = 'start'
df.loc[stop, 'flag'] = 'stop'

#    value   flag
# 0    NaN    NaN
# 1    NaN    NaN
# 2    1.0  start
# 3    2.0    NaN
# 4    1.0    NaN
# 5    3.0    NaN
# 6    2.0   stop
# 7    NaN    NaN
# 8    1.0  start
# 9    2.0   stop

或者使用以下方式分配mask：

df['flag'] = df['flag'].mask(start, 'start')
df['flag'] = df['flag'].mask(stop, 'stop')