如何在没有多个 for 循环的情况下从 nd numpy 数组中删除一对反向元素？

Question

例如：

原文：[[0 2] [0 3] [1 4] [2 0] [2 3] [3 0] [3 2] [4 1]]

编辑：[[0 2] [0 3] [1 4] [2 3]]

Answer 1

一种选择：排序并获取唯一值：

a = np.array([[0, 2], [0, 3], [1, 4], [2, 0], [2, 3], [3, 0], [3, 2], [4, 1]])

out = np.unique(np.sort(a), axis=0)

output：

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

如果要确保保留原始（未排序）数据：

# different example, note the initial [2, 0]
a = np.array([[2, 0], [0, 3], [1, 4], [2, 0], [2, 3], [3, 0], [3, 2], [4, 1]])

idx = np.unique(np.sort(a), axis=0, return_index=True)[1]

a[idx]

output：

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

Answer 2

一种快速简单的方法，带有一个for循环：

s = [[0, 2], [0, 3], [1, 4], [2, 0], [2, 3], [3, 0], [3, 2], [4, 1]]
 
result = []
result.append(s[0])

for x in s:
    if not([x[-1],x[0]] in result) and not(x  in result):
        result.append(x)
result

Output：

[[0, 2], [0, 3], [1, 4], [2, 3]]