在可变深度处切片嵌套列表

时间:2014-12-07 03:10:15

标签: python

如何为变量' depth'?

的任意值实现以下内容
if depth == 1:
    txt[0] = txt[0][s:]
if depth == 2:
    txt[0][0] = txt[0][0][s:]
if depth == 3:
    txt[0][0][0] = txt[0][0][0][s:]

1 个答案:

答案 0 :(得分:1)

您可以使用循环:

target = txt
for _ in range(depth - 1):
    target = target[0]
target[0] = target[0][s:]

这反复遍历下一个第一个元素;最后一个级别然后被切片替换。