Python 交换函数

我很难用 Python 表达这一点。

这是需要做什么的描述。

swap_cards: (int, int 列表) -> NoneType

swap_cards([3, 2, 1, 4, 5, 6, 0], 5)
[3, 2, 1, 4, 5, 0, 6]

swap_cards([3, 2, 1, 4, 5, 6, 0], 6)
[0, 2, 1, 4, 5, 6, 3]`

我创建了 2 个示例，但我不知道如何启动函数体。

听起来这里需要一些索引符号：

>>> def swap_cards(L, n):
...     if len(L) == n + 1:
...         L[n], L[0] = L[0], L[n]
...         return L
...     L[n], L[n+1] = L[n+1], L[n]
...     return L
... 
>>> swap_cards([3, 2, 1, 4, 5, 6, 0], 5)
[3, 2, 1, 4, 5, 0, 6]
>>> swap_cards([3, 2, 1, 4, 5, 6, 0], 6)
[0, 2, 1, 4, 5, 6, 3]