当前,我有一个NumPy数组:
a = np.array([[1,2],[3,4],[5,6]])
并且我正在itertools中进行产品操作,但是我想添加一个步进函数。例如:
[ x for x in itertools.product(*a) ]
结果:
[(1, 3, 5), (1, 3, 6), (1, 4, 5), (1, 4, 6), (2, 3, 5), (2, 3, 6), (2, 4, 5), (2, 4, 6)]
对于步骤(1,1,2)的目标输出:
[(1, 3, 5), (1, 4, 5), (2, 3, 5), (2, 4, 5)]
我不知道,这是问题所在。
itertools
工具来执行此操作。In [586]: list(itertools.islice(itertools.product(*a), None, None,2))
Out[586]: [(1, 3, 5), (1, 4, 5), (2, 3, 5), (2, 4, 5)]