我有興趣在至少三個(gè)相同數(shù)字的序列上找到第一個(gè)和最后一個(gè)索引（它們必須排成一行）。如果有更多序列，索引（開始-結(jié)束）將附加到列表中。例子： s = [1,1,1,1,4,1,1,1] --> output: [0,3,5,7]s = [1,1,1,1,4,1,1,1]c = []indexes = []for i in range(len(s)): if s.count(s[i]) >= 3: c.append(i)my output: [0, 1, 2, 3, 5, 6, 7]Python順序
2 回答
呼如林
TA貢獻(xiàn)1798條經(jīng)驗(yàn) 獲得超3個(gè)贊
你可以使用groupby.?讓我們從以下內(nèi)容開始：
s = [1, 1, 1, 1, 4, 1, 1, 1]
for value, group in itertools.groupby(s):
? ? # print(value)
? ? print(list(group))
這會(huì)給你
[1, 1, 1, 1]
[4]
[1, 1, 1]
現(xiàn)在讓我們添加您的條件并跟蹤當(dāng)前位置。
s = [1, 1, 1, 1, 4, 1, 1, 1]
positions = []
current_position = 0
for value, group in itertools.groupby(s):
? ? group_length = len(list(group))
? ? if group_length >= 3:
? ? ? ? positions.extend([current_position, current_position + group_length - 1])
? ? current_position += group_length
print(positions)
這會(huì)給你想要的結(jié)果[0, 3, 5, 7]。
慕絲7291255
TA貢獻(xiàn)1859條經(jīng)驗(yàn) 獲得超6個(gè)贊
在這里，嘗試使用此代碼來解決您的問題：
prev_value = s[0]
prev_index = 0
consecutive_count = 0
for index, value in enumerate(s):
if value == prev_value:
consecutive_count += 1
else:
if consecutive_count > 2:
indexes.append(prev_index)
indexes.append(index - 1)
consecutive_count = 1
prev_value = value
prev_index = index
if consecutive_count > 2:
indexes.append(prev_index)
indexes.append(index)
添加回答
舉報(bào)
0/150
提交
取消