У меня есть серия, как показано ниже. Серия представляет собой фрейм данных. Я конвертирую в серию только для того, чтобы показать здесь.
Responses = {N, N, N, N, N, S, N, N, N, N, N, N, S, N, N, N, N, N, N, N, N, N, N, N, N, N, S, S, S, S, S, S, N, N, S, N, N, N, S, S, S, S, N, S, N, N, N, N, N, N, N, S, S, N, N, N, S, S, S, N, S, S, S, S, S, S, S, S,N, N, N, N, N, S, N, N, N, N, N, N, S, N, N, N, N, N, N, N, N, N, N, N, N, N, S, S, S, S, S, S, N, N, S, N, N, N, S, S, S, S, N, S, N, N, N, N, N, N, N, S, S, N, N, N, S, S, S, N, S, S, S, S, S, S, S, S }
Я использую этот код для подсчета наибольшей последовательности «S» (я знаю, что это не лучший способ):
import csv
import numpy as np
import pandas as pd
df = pd.read_csv('C:\\teste\\pergunta.csv')
count = 0
prev = 0
indexend = 0
indexcount = 0
for i in range(0,len(df)):
if df['Responses'].loc[i] == 'S':
count += 1
indexcount = i
else:
if count > prev:
prev = count
indexend = i
count = 0
if count > prev:
prev = count
indexend = indexcount
print("The longest sequence is "+str(prev))
count = 0
prev = 0
indexend = 0
indexcount = 0
Код говорит, что самая длинная последовательность os 'S' равна 8. Это нормально.
Но эта последовательность из 8 S происходит дважды.
Мой вопрос: мне нужен код, который считает отдельные последовательности S от самой длинной до минимальной.
Что-то вроде:
Кто-нибудь может мне помочь?
Спасибо
Пытаться:
import pandas as pd
from itertools import groupby
out = {}
for k, g in groupby(Responses):
if k == 'S':
l = sum(1 for _ in g)
out[l] = out.get(l, 0) + 1
df = pd.DataFrame([{'Sequence': k, 'Frequency': f'{v} times'} for k, v in out.items()]).sort_values('Sequence', ascending=False)
print(df)
Отпечатки:
Sequence Frequency
5 8 2 times
1 6 2 times
2 4 2 times
4 3 2 times
3 2 2 times
0 1 8 times
Попробуй это:
s = pd.Series(Responses)
s = s.loc[s.eq('S')].groupby(s.ne(s.shift()).cumsum()).count().value_counts()
df = s.map('{} times'.format).rename_axis('Sequence').reset_index(name = 'Frequency')
Выход:
Sequence Frequency
0 1 8 times
1 6 2 times
2 4 2 times
3 2 2 times
4 3 2 times
5 8 2 times
Использование itertools
:
import itertools as it
import pandas as pd
N='N'
S='S'
df = pd.DataFrame({'Responses':[N, N, N, N, N, S, N, N, N, N, N, N, S, N, N, N, N, N, N, N, N, N, N, N, N, N, S, S, S, S, S, S, N, N, S, N, N, N, S, S, S, S, N, S, N, N, N, N, N, N, N, S, S, N, N, N, S, S, S, N, S, S, S, S, S, S, S, S,N, N, N, N, N, S, N, N, N, N, N, N, S, N, N, N, N, N, N, N, N, N, N, N, N, N, S, S, S, S, S, S, N, N, S, N, N, N, S, S, S, S, N, S, N, N, N, N, N, N, N, S, S, N, N, N, S, S, S, N, S, S, S, S, S, S, S, S] })
l = [list(g).count('S') for k, g in it.groupby(df['Responses']) if k=='S']
r = (pd.DataFrame.from_dict({i:l.count(i) for i in range(min(l), max(l)+1)},
orient='index')
.reset_index()
)
r.columns=['Sequence', 'Frequency']
print(r)
Sequence Frequency
0 1 8
1 2 2
2 3 2
3 4 2
4 5 0
5 6 2
6 7 0
7 8 2