python - python itertools.imap(func,p,q) 可以对索引进行条件测试（例如，仅获取具有奇数索引的元素）吗？

Question

我有以下数组

>>> a
array([0, 8, 0, 8, 0, 8])
>>> b
array([0, 6, 0, 6, 0, 6])

这些代表一组复数的实部和虚部。

我可以numpy.complex使用以下方法将它们重新格式化为数据类型

>>> [x for x in itertools.imap(complex,a,b)]
[0j, (8+6j), 0j, (8+6j), 0j, (8+6j)]

但是我真正想要得到的只是具有奇数索引的元素：

[(8+6j),(8+6j),(8+6j)]

有没有简单的方法来实现这一目标？

Answer 1

>>> import numpy as np
>>> a = np.array([0, 8, 0, 8, 0, 8])
>>> b = np.array([0, 6, 0, 6, 0, 6])
>>> np.vectorize(complex)(a, b)[1::2]
array([ 8.+6.j,  8.+6.j,  8.+6.j])

那将是最好的方法，但为了完整性，itertools解决方案是：

>>> from itertools import imap, islice
>>> list(islice(imap(complex,a,b), 1, None, 2))
[(8+6j), (8+6j), (8+6j)]

Answer 2

[1::2]从索引 1（第二个）开始，将所有其他项目带到列表的末尾。那是：

>>> [x for x in itertools.imap(complex,a,b)][1::2]

[(8+6j), (8+6j), (8+6j)]