比较 2 个元组列表

Question

我有一个清单：

 a = [(1, 2), (3, 4), (4, 5), (6, 7)]
 # Stores list of x,y coordinates

和一个清单：

 b = [(1, 2), (10, 1), (3, 10), (4, 9)]

现在，我想用 y 坐标 + 2 替换a >= of, b中a y 坐标。

因为这里a有一个等价或更大的b在：

[(1,2), (3,4)]

我想替a这样：

a = [(1,4), (3,6), (4,5), (6,7)]

我怎么能这样做？

我知道 numpy 存在一种方法，例如：

np.where(a >= b) , do something;

但不确定在这种情况下如何使用它。

Answer 1

IIUC，比较他们的axis=1和+=2

a = np.asarray(a)
b = np.asarray(b)

a[a[:, 1] > b[:, 1], 1] += 2

---

array([[1, 2],
       [3, 6],
       [4, 5],
       [6, 7]])

Answer 2

没有 numpy：

a = [(1, 2), (3, 4), (4, 5), (6, 7)]
b = [(1, 2), (10, 1), (3, 10), (4, 9)]
c = [(aa[0], aa[1]+2) if aa[1] >= bb[1] else aa for aa, bb in zip(a, b)]

c是[(1, 4), (3, 6), (4, 5), (6, 7)]