元素列表比较

Question

我有一个问题，我很难解释，所以我会用很多例子来帮助大家理解，看看你是否能帮助我。

假设我有两个列表，其中包含两个人从最好到最差的书名。用户1评价lstA，用户2评价lstB

lstA = ['Harry Potter','1984','50 Shades','Dracula']
lstB = ['50 Shades','Dracula','1984','Harry Potter']

用户一认为“哈利波特”比“德古拉”好（HP为索引0，德古拉为索引3）

用户二认为“哈利波特”比德古拉差，（HP是索引3，德古拉是索引1）

在这种情况下，返回一个元组('Harry Potter', 'Dracula') [('Dracula', 'Harry Potter')也可以]

用户 1 的“50 种色调”也比“Dracula”好，用户 2 的“50 种色调”也比“Dracula”好（指数分别为 2、3 和 0、1）。在这种情况下，什么都不会发生。

程序的最终结果应该返回一个元组列表，

[('Harry Potter','50 Shades'), ('Harry Potter','Dracula'), ('Harry Potter','1984'), ('1984', '50 Shades'), ('1984','Dracula')]

有人可以帮我指出正确的方向，以提出一种给出所有元组的算法吗？

Answer 1

首先以数学方式制定您的逻辑。对于长度为 2 的所有组合，给定索引 idx_a1, idx_a2 和 idx_b1, idx_b2，如果是 sign(idx_a1 - idx_a2) != sign(idx_b1 - idx_b2)，则记录组合。

以下内容效率不高，但它显示了将此逻辑转换为代码的一种方法：

from itertools import combinations

lstA = ['Harry Potter','1984','50 Shades','Dracula']
lstB = ['50 Shades','Dracula','1984','Harry Potter']

def sign(x):
    """Return +1 if integer is positive, -1 if negative"""
    return (x > 0) - (x < 0)

res = []
for a, b in combinations(lstA, 2):
    idx_a1, idx_a2 = lstA.index(a), lstA.index(b)
    idx_b1, idx_b2 = lstB.index(a), lstB.index(b)
    if sign(idx_a1 - idx_a2) != sign(idx_b1 - idx_b2):
        res.append((a, b))

[('Harry Potter', '1984'),
 ('Harry Potter', '50 Shades'),
 ('Harry Potter', 'Dracula'),
 ('1984', '50 Shades'),
 ('1984', 'Dracula')]

Answer 2

这样做的一种方法是将每个列表中的所有正序累积到一个集合中，然后取这两个集合的差。当a 在其各自列表中位于b 之前时，正序将是(a, b)。这是itertools.combinations保证的顺序：

from itertools import combinations

setA = set(combinations(lstA, 2))
setB = set(combinations(lstB, 2))

result = setA - setB

这将简单地丢弃两组同意的任何顺序。如果两个列表有相同的书，这将几乎相同

result = setB - setA

唯一的区别是所有元组都会被反转。

如果您在每个列表中有不同的书籍，则需要添加几个额外的步骤来清理重复项并合并两组：

resultA = setA - setB
resultB = setB.difference(x[::-1] for x in setA)
result = resultA | resultB

第一步计算lstA 中lstB 不同意的所有元素。下一步找到lstB 的元素，这些元素不是我们在resultA 中的反转版本，因为两个列表中书籍的分歧保证在集合中被反转。我在这里使用set.difference 方法而不是- 运算符，因为这样就不需要从生成器表达式创建集合对象。不幸的是，您不能只使用symmetric_difference/^，因为这些元素是相反的。第三步只是计算结果的并集。

IDEOne 链接：https://ideone.com/DuHTed。这演示了问题中的原始案例和非对称列表。

Answer 3

@jpp 解决方案的一个高效版本如下：

from itertools import combinations

lstA = ['Harry Potter','1984','50 Shades','Dracula']
lstB = ['50 Shades','Dracula','1984','Harry Potter']

bIndices = {b: i for i, b in enumerate(lstB)}
aPairs = [sorted(c) for c in combinations(enumerate(lstA), 2)]

mismatches = [(book1[1], book2[1]) for book1, book2 in aPairs if bIndices[book1[1]] > bIndices[book2[1]]]
print(mismatches)
# [('Harry Potter', '1984'), ('Harry Potter', '50 Shades'), ('Harry Potter', 'Dracula'), ('1984', '50 Shades'), ('1984', 'Dracula')]

请注意，aPairs 是 (index, book) 元组的组合，并且每个组合都按索引排序，这保证在每对书籍中，第一个比下一个“更好”（对于用户 A）。

现在要计算排序不匹配，我们只需要确定lstB 中的相应书籍索引是否也保留此排序。

编辑

正如@MadPhysicist 所指出的，combinations 在每个生成的元组中保留数组中的原始顺序，因此无需将aPairs 创建为已排序(index, book) 元组的列表。我们可以直接用bIndices生成mismatches：

lstA = ['Harry Potter','1984','50 Shades','Dracula']
lstB = ['50 Shades','Dracula','1984','Harry Potter']

bIndices = {b: i for i, b in enumerate(lstB)}
mismatches = [(book1, book2) for book1, book2 in combinations(lstA, 2) if bIndices[book1] > bIndices[book2]]

Answer 4

你可以使用iter然后比较索引

res = []  

for i in lstA:
    a = iter(lstB)
    while True:
        try:
            b = next(a)
            if lstA.index(i) < lstA.index(b) and lstB.index(i) > lstB.index(b):
                res.append((i, b))
        except StopIteration:
            break 

print(res)
# [('Harry Potter', '50 Shades'), ('Harry Potter', 'Dracula'), ('Harry Potter', '1984'), ('1984', '50 Shades'), ('1984', 'Dracula')]