我想将整数数组传递给itemgetter函数。我怎么能这样做?
以下是我的代码:
// Start of code
start ="1,3,5"
mylist = start.split(',')
key = itemgetter(*start)
or
key = itemgetter(mylist)
print key
答案 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')