python - 有几个间隔的切片？

Question

我有一个文件，我想在其中提取第 2、3、4、5 列和第 -4 列。这些列不相邻。

出于代码整洁的原因，我想做类似的事情

values = line.split()[columns_to_extract]

代替


values_part_one = line.split()[columns_to_extract_one]
values_part_two = line.split()[columns_to_extract_two]


因此，我想制作一个包含位置 2、3、4、5 和 -4 的切片，以便能够在一行中提取值。这可能吗？

如果没有，是否还有其他整洁的 oneliners 可以做到这一点？

Answer 1

是否可以制作切片来做到这一点？没有。

但是，一切并没有丢失!您可以使用operator.itemgetter :

getter = operator.itemgetter(2, 3, 4, 5, -4)

例子:

>>> import operator
>>> getter = operator.itemgetter(2, 3, 4, 5, -4)
>>> getter(range(50))  # Note, returns a `tuple`
(2, 3, 4, 5, 46)