Как получить конкретные строки по условиям? - PullRequest
Новая
       23

Как получить конкретные строки по условиям?

0 голосов
/ 12 ноября 2018

Для данных следующим образом: 

Name      Stage           Start                 End

Hulk        1      21/10/2018 06:34:15    21/10/2018 07:34:15
Hulk        2      21/10/2018 07:34:15    21/10/2018 07:54:15
Hulk        3      21/10/2018 07:58:15    21/10/2018 08:14:15
Hulk        4      21/10/2018 08:14:15    21/10/2018 08:34:15
Sam         A1     21/10/2018 09:34:15    21/10/2018 10:34:15
Sam         A2     21/10/2018 10:34:15    21/10/2018 10:45:15
Sam         A3     21/10/2018 10:45:15    21/10/2018 11:00:15
Sam         A4     21/10/2018 11:00:15    21/10/2018 11:34:15
Bruce       1.1    21/10/2018 11:34:15    21/10/2018 11:45:15
Bruce       1.2    21/10/2018 11:45:15    21/10/2018 12:00:15
Bruce       1.3    21/10/2018 12:00:15    21/10/2018 12:25:15
Bruce       1.4    21/10/2018 12:25:15    21/10/2018 12:45:15
Peter        1     21/10/2018 12:45:15    21/10/2018 01:05:15
Peter        1     21/10/2018 01:05:15    21/10/2018 01:15:15

Как я могу иметь first и last экземпляр Stage для каждого Name типа, которые начинаются с 1 в нем и продолжаются с 4?

Кадр данных должен быть следующим: 

Name      Stage           Start                 End

Hulk        1      21/10/2018 06:34:15    21/10/2018 07:34:15
Hulk        4      21/10/2018 08:14:15    21/10/2018 08:34:15
Sam         A1     21/10/2018 09:34:15    21/10/2018 10:34:15
Sam         A4     21/10/2018 11:00:15    21/10/2018 11:34:15
Bruce       1.1    21/10/2018 11:34:15    21/10/2018 11:45:15
Bruce       1.4    21/10/2018 12:25:15    21/10/2018 12:45:15

Я пытался groupby с ([Name,Stage]), но не получил желаемый кадр данных, как указано выше.

1 Ответ

0 голосов
/ 12 ноября 2018

Используйте duplicated с str.contains с boolean indexing, чтобы сначала вернуть необходимые строки, а затем value_counts сmap только для фильтра 2 группы строк: 

m1 = ~df['Name'].duplicated()
m2 = df['Stage'].str.contains('1')

m3 = ~df['Name'].duplicated(keep='last')
m4 = df['Stage'].str.contains('4')

df1 = df[(m1 & m2) | (m3 & m4)].copy()

df1 = df1[df1['Name'].map(df1['Name'].value_counts()) == 2]
print (df1)
     Name Stage                Start                  End
0    Hulk     1  21/10/2018 06:34:15  21/10/2018 07:34:15
3    Hulk     4  21/10/2018 08:14:15  21/10/2018 08:34:15
4     Sam    A1  21/10/2018 09:34:15  21/10/2018 10:34:15
7     Sam    A4  21/10/2018 11:00:15  21/10/2018 11:34:15
8   Bruce   1.1  21/10/2018 11:34:15  21/10/2018 11:45:15
11  Bruce   1.4  21/10/2018 12:25:15  21/10/2018 12:45:15
...