python - 在python中按索引切片嵌套列表

Question

我有一个不等长的嵌套列表:[[1,2,3],[4,5],[6,7,8]] 并且我有一个 start_index=(i,j) 和 end_index=(a,b) 并且我需要打印 start_index 和 end_index 之间的所有元素。例如，如果 start_index=(1,1) 和 end_index=(2,2) 那么我将打印 (5,6,7,8)

Answer 1

您可以使用以下功能:

def nested_index(arr, start, end):
    res = arr[start[0]][start[1]:]
    for i in range(start[0] + 1, end[0]):
        res.extend(arr[i])
    res.extend(arr[end[0]][:end[1] + 1])
    return res

>>> print(nested_index([[1,2,3],[4,5],[6,7,8]], (1, 1), (2, 2)))
[5, 6, 7, 8]