如何在项getter函数中传递整数数组作为参数?

时间:2016-01-25 16:34:51

标签: python csv

我想将整数数组传递给itemgetter函数。我怎么能这样做?

以下是我的代码:

// Start of code

start ="1,3,5"
mylist = start.split(',')
key = itemgetter(*start)
or
key = itemgetter(mylist)
print key

1 个答案:

答案 0 :(得分:2)

你几乎有正确的想法。如果您尝试索引可迭代的内容,例如一个列表,那么这些项应该是整数而不是字符串:

>>> start = "1,3,5"
>>> mylist = [int(x) for x in start.split(',')]
>>> g = itemgetter(*mylist)
>>> g(range(10))
(1, 3, 5)
>>> g('abcdefghi')
('b', 'd', 'f')