[已解决]list中寻找连续出现的位置

xzc_98

求教，在一个list中，想找到连续2次或3次或4次（连续出现次数可根据输入值变化）True的位置。
例如a = [False, True, True, True, False, True, True, True, True, False] 。这个list的答案应该是[[1, 3],[5,4]]。
或者[False] * 3。答案应该是[]。
再或者[True, True, False, False, True, True, True, True, True, False, False, True, True] 。答案应该是[[0, 2],[4,5],[11,2]]。
该怎样编写？

最佳答案

月排行榜 / 总排行榜

qq1151985918

2021-6-28 09:05:56

1. def fun(data: list) -> list:
   
2.     results, result = [], [None, 0]
   
3.     for i in range(len(data)):
   
4.         if data[i]:
   
5.             if result[0] is None:
   
6.                 result[0] = i
   
7.                 result[-1] += 1
   
8.             else:
   
9.                 result[-1] += 1
   
10.         else:
    
11.             if result[0] is not None and result[-1] >= 2:
    
12.                 results.append(result)
    
13.             result = [None, 0]
    
14.     else:
    
15.         if result[0] is not None and result[-1] >= 2:
    
16.             results.append(result)
    
17.     return results
    
18. 
    
19. 
    
20. if __name__ == "__main__":
    
21.     a = [True, True, False, False, True, True, True, True, True, False, False, True, True]
    
22.     print(fun(a))
    

复制代码

跳转到最佳答案楼层

qq1151985918

1. def fun(data: list) -> list:
   
2.     results, result = [], [None, 0]
   
3.     for i in range(len(data)):
   
4.         if data[i]:
   
5.             if result[0] is None:
   
6.                 result[0] = i
   
7.                 result[-1] += 1
   
8.             else:
   
9.                 result[-1] += 1
   
10.         else:
    
11.             if result[0] is not None and result[-1] >= 2:
    
12.                 results.append(result)
    
13.             result = [None, 0]
    
14.     else:
    
15.         if result[0] is not None and result[-1] >= 2:
    
16.             results.append(result)
    
17.     return results
    
18. 
    
19. 
    
20. if __name__ == "__main__":
    
21.     a = [True, True, False, False, True, True, True, True, True, False, False, True, True]
    
22.     print(fun(a))
    

复制代码

阿奇_o

1. # 思路：将False看做是括号，那么两个False之间，若为连续的True就可以很好地判断了
   
2. 
   
3. # 代码实现：
   
4. a = [False, True, True, True, False, True, True, True, True, False]
   
5. # a = [True, True, True, False, True, False, True, True, True]  # 测试
   
6. a = [True, True, False, False, True, True, True, True, True, False, False, True, True]
   
7. 
   
8. # 前后插入False，确保 True被False“包住”
   
9. a.insert(0, False)
   
10. a.append(False)
    
11. 
    
12. # 转为01字符串，以便于后面写判断
    
13. s = ''.join([str(int(i)) for i in a])  
    
14. print(s)
    
15. 
    
16. # 确定False（'0'）出现的位置
    
17. fs = [i for i in range(len(s)) if s[i]=='0']  
    
18. print(fs)
    
19. 
    
20. # 根据前后两个False出现的位置，判断其之间 是否包着 连续的True（'1')
    
21. lx = ['1'*(n+1) for n in range(int(input("True的最大连续次数为：")))]
    
22. result = []
    
23. for stop in fs:
    
24.     for start in fs:
    
25.         ss = s[int(start) + 1 :int(stop)]
    
26.         if ss in lx:  # 如 '11', '111', '1111'
    
27.             # print([start, ss])
    
28.             result.append([start, len(ss)])   
    
29.             # 注：正确的开始位置 本应该是 start+1，但我们一开始在 insert()的False，所以这里抵消了， 故是 start
    
30. 
    
31. print(result)

复制代码

修改：有点偏差局限了，需改一下 去掉lx不用； 将 if ss in lx:  改为  if '0' not in ss and len(ss) >=2 :

OK。

xzc_98

qq1151985918 发表于 2021-6-28 09:05

你好，如果我想改变原来2个连续就计入变为3个连续才计入，该怎么做？

qq1151985918

xzc_98 发表于 2021-6-28 23:25
你好，如果我想改变原来2个连续就计入变为3个连续才计入，该怎么做？

代码里的所有2改成3就好了

xzc_98

qq1151985918 发表于 2021-6-28 23:27
代码里的所有2改成3就好了

谢谢！

pppppddd

还有别的方法不

pppppddd

这刚开始result这步没看懂啊