如何在Python中查找字符的开始和结束事件

2019年7月28日 163次阅读

我有一个带有以下ID的数据帧df(在Col中).

A / B / C的最后一次出现代表开始,X的最后一次出现是结束.我应该忽略开始和结束之间的任何其他A,B,C(例如,行8和9).

我必须从这些数据中找到开始和结束记录,并为每个事件分配一个数字.列数是我想要的输出：

    Col         ID
    P
    Q
    A             
    A              
    A             1
    Q             1 
    Q             1 
    B             1 
    C             1
    S             1 
    S             1
    X             1
    X             1  
    X             1 
    Q
    Q
    R
    R
    C              
    C             2 
    D             2 
    E             2 
    B             2 
    K             2 
    D             2 
    E             2 
    E             2 
    X             2 
    X             2

这段代码：

lc1 = df.index[df.Col.eq('A') & df.Col.ne(df.Col.shift(-1))]

会给我一个最后出现的所有索引值’A’的数组,在本例中为[5].

lc1 = df.index[df.Col.eq('C') & df.Col.ne(df.Col.shift(-1))]  # [20]
lc2 = df.index[df.Col.eq('X') & df.Col.ne(df.Col.shift(-1))]  # [14,29]

我会用iloc打印计数值：

df.iloc[5:14]['count'] = 1
df.iloc[20:29]['count'] = 2

如何找到A / B / C的索引并打印每个开始和结束事件的计数值？

最佳答案 要查找A,B和C的索引,您可以：

df[(df.Col =='A')|(df.Col =='B')|(df.Col =='C')].index

打印您的起始计数：

df1 = df[df['count'] != df['count'].shift(+1)]
print df1[df1['count'] != 0]['count']

打印您的最终计数：

df2 = df[df['count'] != df['count'].shift(-1)]
print df2[df2['count'] != 0]['count']

在旁注中,调用列数是一个坏主意,因为count是DataFrame的一种方法,然后在执行df.count时会出现歧义.

编辑：纠正,因为我回答了一个错误的问题.

点赞