有没有办法获取列表的特定索引,就像我在 NumPy 中可以做的那样?
sample = ['a','b','c','d','e','f']
print sample[0,3,5]
>>>['a','d','f']
我试过用谷歌搜索这个,但我找不到一个好的方法来表达我的问题,导致相关结果......
有没有办法获取列表的特定索引,就像我在 NumPy 中可以做的那样?
sample = ['a','b','c','d','e','f']
print sample[0,3,5]
>>>['a','d','f']
我试过用谷歌搜索这个,但我找不到一个好的方法来表达我的问题,导致相关结果......
您可以使用列表推导:
>>> sample = ['a','b','c','d','e','f']
>>> [sample[i] for i in (0, 3, 5)]
['a', 'd', 'f']
或者,我很快做了一些事情:
>>> class MyList(list):
... def __getitem__(self, *args):
... return [list.__getitem__(self, i) for i in args[0]]
...
>>> mine = MyList(['a','b','c','d','e','f'])
>>> print mine[0, 3, 5]
['a', 'd', 'f']