跟踪计算的子集以避免混合结果

Question

棘手的名称和不清楚的名称...

我的问题如下：

import itertools
t0 = 0
tf = 1000000
# inputs_to_compute = list-like of size 2 to 6 of objects
results = [[] for i in range(len(inputs_to_compute))]
for subset in itertools.combinations(inputs_to_compute, 2):
r1, r2 = compute(subset[0], subset[1], t0, tf)
results[inputs_to_compute.index(subset[0])] += list(r1)
results[inputs_to_compute.index(subset[1])] += list(r2)

此代码创建的结果列表与输入一样多。每个输入实际上都与一个列表相关联。然后执行 2 x 2(在每个子集上(的计算，并将结果添加到相应的列表中。

只要输入中没有重复，它就可以很好地工作，因为该方法index返回元素的第一次出现。我怎样才能以不同的方式实现这一点(并且有效地，性能是我遇到的主要问题之一(来管理重复项？

虚拟示例：

import itertools
def compute(x, y):
return (x + y, x - y)
inputs_to_compute = [1, 1, 3]
results = [[] for i in range(len(inputs_to_compute))]
for subset in itertools.combinations(inputs_to_compute, 2):
r1, r2 = compute(subset[0], subset[1])
results[inputs_to_compute.index(subset[0])].append(r1)
results[inputs_to_compute.index(subset[1])].append(r2)

输出：

[[2, 0, 4, 4], [], [-2, -2]]

预期产出：

# Iteration (1, 1): r1 = 2, r2 = 0
results = [[2], [0], []]
# Iteration (1, 3): r1 = 4, r2 = -2
results = [[2, 4], [0], [-2]]
# Iteration (1, 3): r1 = 4, r2 = -2
results = [[2, 4], [0, 4], [-2, -2]]

Answer 1

for subset_with_indices in itertools.combinations(enumerate(inputs_to_compute), 2):
i1,x1 = subset_with_indices[0]
i2,x2 = subset_with_indices[1]
r1, r2 = compute(x1, x2)
results[i1].append(r1)
results[i2].append(r2)

Answer 2

如果我理解正确，那么每次重复的1作为subset的一部分返回时，您都希望在重复的之间循环。

执行此操作的一种方法是创建一个字典，其中项作为键，其索引存储在列表中。一旦我们有了这个字典，我们就可以将itertools.cycle应用于列表，然后使用next()在项目的索引之间循环：

import itertools

def compute(x, y):
return (x + y, x - y)
inputs_to_compute = [1, 1, 3]
indices = {}
for ind, item in enumerate(inputs_to_compute):
indices.setdefault(item, []).append(ind)
for k, v in indices.items():
indices[k] = itertools.cycle(v)
results = [[] for i in range(len(inputs_to_compute))]
for subset in itertools.combinations(inputs_to_compute, 2):
r1, r2 = compute(subset[0], subset[1])
results[next(indices[subset[0]])].append(r1)
results[next(indices[subset[1]])].append(r2)

输出：

>>> %run so.py
>>> results
[[2, 4], [0, 4], [-2, -2]]

Answer 3

在文档中，您有一个使用排列的组合配方[1]。 您只需修改它以返回索引

import itertools
def compute(x, y):
return (x + y, x - y)
def combinations(iterable, r):
pool = tuple(iterable)
n = len(pool)
for indices in itertools.permutations(range(n), r):
if sorted(indices) == list(indices):
yield tuple(indices)
inputs_to_compute = [1, 1, 3]
results = [[] for i in range(len(inputs_to_compute))]
for i1, i2 in combinations(inputs_to_compute, 2):
r1, r2 = compute(inputs_to_compute[i1], inputs_to_compute[i2])
results[i1].append(r1)
results[i2].append(r2)
print(results)

[1] https://docs.python.org/3/library/itertools.html#itertools.combinations