Question

python 如果数组里含有需要的数时，取其下一数据

[1,3,8,21,30,32]
[5,6,8,15,22,31]
[1,2,3,8,22,25]
[1,3,5,8,22,29]
[4,5,15,22,30,31]

含有[1,3,8]时，把下一行的得出来：[1,3,8],[5,6,8,15,22,31],[1,3,5,8,22,29],[4,5,15,22,30,31]

含有[5,8,22]时，把下一行的得出来: [5,8,22],[1,2,3,8,22,25],[4,5,15,22,30,31]
求代码，谢谢

Answer 1

aa=[[1,3,8,21,30,32],[5,6,8,15,22,31],[1,2,3,8,22,25],[1,3,5,8,22,29],[4,5,15,22,30,31]]
arr1=[[1,3,8]]
arr2=[[5,8,22]]
for x in range(len(aa)-1):
if set(arr1[0]).issubset(set(aa[x])):
arr1.append(aa[x+1])
if set(arr2[0]).issubset(set(aa[x])):
arr2.append(aa[x+1])
print(arr1,'\n'*3,arr2)

Answer 2