python - 从变量索引开始循环遍历列表

Question

是否有一种单行代码可以替代此代码中的 try 和 except？

在列表中查找变量然后返回下一个索引，如果索引超出范围则返回第一个。

def cycle_my_list(my_var):
    my_list = ['cleaner', 'list', 'cycle']
    try:
        return my_list[my_list.index(my_var) + 1]
    except IndexError:
        return my_list[0]

预期结果:

>>> cycle_my_list('list')
'cycle'
>>> cycle_my_list('cycle')
'cleaner'

Answer 1

使用模 % 运算符

my_list = ['cleaner', 'list', 'cycle']
return my_list[(my_list.index(my_var) + 1) % len(my_list)]