繁体   English   中英

如何从python中的列表索引获取列表的子集

[英]how to get subset of list from index of list in python

我有一个字符串列表,我需要以通用方式根据列表列表从列表中进行子集化

indx=[0,5,7] # index list

a = ['a', 'b', 3, 4, 'd', 6, 7, 8]

我需要在第一次迭代中以通用方式获取子集:

a[0:5]

第二次迭代:

a[5:7]

第3次迭代:

a[7:]

我试过的代码:

for i in indx:
    if len(indx)==i:
        print(a[i:])
    else:
        print(a[i:i+1])

预期产量:

a[0:5]=='a', 'b', 3, 4, 'd'
a[5:7]=6, 7
a[7:]=8

使用理解:

>>> [a[x:y] for x,y in zip(indx,[*indx[1:], None])]
[['a', 'b', 3, 4, 'd'], [6, 7], [8]]
results = [a[indx[i]:indx[i+1]] for i in range(len(indx)-1)]

将返回包含所需3个列表的列表。

你可以试试这个:

indx.append(len(a))
print(*[a[i:j] for i,j in zip(indx, indx[1:])], sep='\n')

输出

['a', 'b', 3, 4, 'd']
[6, 7]
[8]

你正好考虑一次迭代两个项目,但问题是for i in lst中不使用索引,并且会失败。

迭代和获取两个项目的一种方法是使用zip。

indx=[0,5,7] # index list

a = ['a', 'b', 3, 4, 'd', 6, 7, 8]

if indx[0] != 0:
    indx = [0] + indx #fixes left indexing in case 0 is not present
if indx[-1] != len(a):
    indx += [len(a)] #fixes right indexing to make sure you get all values from the list.

print(indx) #[0, 5, 7, 8]
for left, right in zip(indx, indx[1:]):
    print(a[left: right])
#Output:
['a', 'b', 3, 4, 'd']
[6, 7]
[8]

干得好。

for i in range(len(indx)):
try:
    print("a[%s:%s] ==" % (indx[i], indx[i + 1]) + " ", end=" ")
    print(a[indx[i]:indx[i + 1]])
except IndexError:
    print("a[%s:] ==" % (indx[i]) + " ", end=" ")
    print(a[indx[i]:])

输出:

a[0:5] ==  ['a', 'b', 3, 4, 'd']
a[5:7] ==  [6, 7]
a[7:] ==  [8]

您可以使用slice类型将索引列表转换为列表切片。 正如其他人已发布的那样,使用zip创建起始 - 完成对:

>>> print(list(zip(indx, indx[1:]+[None])))
[(0, 5), (5, 7), (7, None)]

我们使用这些对来定义切片:

>>> slices = [slice(a, b) for a, b in zip(indx, indx[1:]+[None])]

然后像使用整数索引一样使用切片对象,但切片将选择切片开始 - 结束对中定义的列表子串:

>>> for slc in slices:
...     print(a[slc])

得到:

['a', 'b', 3, 4, 'd']
[6, 7]
[8]

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM