python - 元素的列表比较

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') 也可以]

用户一也将“50 种色调”评为优于“德古拉”，用户二也将“50 种色调”评为优于“德古拉”(指数分别为 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')]