[英]Slicing nested list by index in python
我有一個長度不等的嵌套列表: [[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)
您可以使用以下 function:
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]
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.