给定一个input任何类型的序列(列表/字符串/范围)如何找到下一个项目input遵循指定项目?
另外,如果该项目不存在或后面没有任何内容,该函数应该返回None
.
我尝试将输入类型转换为列表,然后从列表中查找位置,然后获取下一个项目,但这并不适用于所有输入类型。我写了一些东西,但我知道它不是Pythonic,而且它也会超时。 (代码战挑战:https://www.codewars.com/kata/542ebbdb494db239f8000046/train/python https://www.codewars.com/kata/542ebbdb494db239f8000046/train/python)
我的尝试:
def next_item(xs, item):
xs_list = list(xs)
if item in xs_list:
position = xs_list.index(item)
try:
return xs_list[position+1]
except IndexError:
return None
else:
return None
期望的结果:
next_item([1, 2, 3, 4, 5, 6, 7, 8], 5)
# 6)
next_item(['a', 'b', 'c'], 'd')
# None)
next_item(['a', 'b', 'c'], 'c')
# None)
next_item('testing', 't')
# # 'e')
next_item(iter(range(1, 3000)), 12)
# , 13)