使用索引对列表进行排序，以便轻松删除项目

Question

我面临以下问题。我在 Python 中有多个列表，我希望它们使用某种索引对它们进行排序，以便从其他列表中删除项目。让我进一步解释。

listA_ID = [1,2,3,5,6,7] # integer from 0-250
listA_Var1 = [3.9, 4.7, 2.1, 1.2, 0.15, 0.99]

listB_ID = [2,5,6,7,8,10] # integer from 0-250
listB_Var1 = [0.54, 0.35, 1.19, 2.45, 3.1, 1.75]

>> After Comparison of listA_ID & listB_ID I should end up with the common IDs.
listA_ID = listB_ID = sorted(list(set(listA_ID) & set(listB_ID)))
listA_ID = [2,5,6,7]
listB_ID = [2,5,6,7]

因此，我想从 listA_ID 中删除位于该列表 [0, 2] 位置的元素 [1, 3]，并从 listA_Var1 中删除相同的元素，删除位于相同位置的 [3.9, 2.1] [0, 2]。

同样，我想从 listB_ID 中删除位于该列表 [4, 5] 位置的元素 [8, 10] 和 listB_Var1 中的相同内容，删除相同的 [3.1, 1.75]位置 [4, 5]。

>> and then listA_Var1 & listB_Var1 will become
listA_Var1 = [4.7, 1.2, 0.15, 0.99]
listB_Var1 = [0.54, 0.35, 1.19, 2.45]

关于有效实施该方法的任何想法？根据我大量使用 Matlab 的经验，在比较两个列表之后，我有办法获取不需要的索引，然后将这些索引应用于列表，我得到的是最终列表 listA_Var1 和 listB_Var1。

有什么想法吗？提前致谢！

Answer 1

1.获得路口

有很多方法可以做到这一点。有关详细讨论，请参阅here。正如那里所建议的那样，如果重复项无关紧要（即您的列表要么不包含重复项，要么它们包含但您不关心它们），例如，您可以使用set() 来获取共同价值观：

intersection_A_B = sorted(list(set(listA_ID) & set(listB_ID)))

Alternatively，也可以只将其中一个列表变成一个集合，然后使用intersection()方法，如：

intersection_A_B = list(set(listA_ID).intersection(listB_ID))

相比之下，如果重复很重要或可能造成问题（例如，listA_ID 和 listB_ID 都包含两次值，并且您希望您的交集保留该值的两个列表），而不是使用set() 或intersection()，你可以use list comprehension:

intersection_A_B = [x for x in listA_ID if x in listB_ID]

---

2。删除值

编辑：得到交叉点后（注意，现在我得到了你真正想要的东西，过程的第一步指的是intersection_A_B，而不是更新listA_ID和@987654335 @ 因为以下操作需要它们的原始状态），这应该可以解决问题：

del_indices_A = [i for i, value in enumerate(listA_ID) if value not in intersection_A_B]
listA_Var1 = [listA_Var1[x] for x in range(len(listA_Var1)) if x not in del_indices_A]

del_indices_B = [i for i, value in enumerate(listB_ID) if value not in intersection_A_B]
listB_Var1 = [listB_Var1[x] for x in range(len(listB_Var1)) if x not in del_indices_B]

这首先检查listA_ID 和listB_ID 中的哪些索引对应于intersection_A_B 中未包含的值，然后排除与listA_Var1 和listB_Var2 中的索引对应的值。

Answer 2

首先，我将逐步解释这种方法：

- 第 1 步： 我们将在 listA_ID 和 listB_ID 中寻找交集元素。

intersection_AB = set(listA_ID) & set(listB_ID)

- 第 2 步： 然后，我们执行 difference of sets。将set(listA_ID)放在首位非常重要，因为the difference of sets is not commutative。

# You can use difference() method alternatively:
# A_elements = list(set(listA_ID).difference(intersection_AB)) but personally I like the minus operator.

A_elements = list(set(listA_ID) - intersection_AB) 

- 第 3 步： 然后，我们根据上一步中找到的元素查找索引。

index_to_remove_list_A = [listA_ID.index(i) for i in A_elements]

或者你也可以使用（虽然不太清晰）：

index_to_remove_list_A = [listA_ID.index(i) for i in list(set(listA_ID) - intersection_AB)]

- 第 4 步： 删除列表中正确的元素。

for i in sorted(index_to_remove_list_A, reverse=True):
  del listA_Var1[i]

print(listA_Var1)

编辑：包含两个列表的完整代码...

A_elements = list(set(listA_ID) - intersection_AB)
B_elements = list(set(listB_ID) - intersection_AB)

index_to_remove_list_A = [listA_ID.index(i) for i in A_elements]
index_to_remove_list_B = [listB_ID.index(i) for i in B_elements]

for i in sorted(index_to_remove_list_A, reverse=True):
  del listA_Var1[i]

for i in sorted(index_to_remove_list_B, reverse=True):
  del listB_Var1[i]


print(listA_Var1) # [4.7, 1.2, 0.15, 0.99]
print(listB_Var1) # [0.54, 0.35, 1.19, 2.45]

Answer 3

好吧，我自己也会发布一个可行的解决方案，使用 numpy。

intersection_A_B = sorted(list(set(listA_ID) & set(listB_ID)))

# Convert Lists to Arrays
np_listA_ID = np.asarray( listA_ID )
np_listB_ID = np.asarray( listB_ID )

# Comparison of two arrays
np_list_ID, listA_ind, listB_ind = np.intersect1d(np_listA_ID, np_listB_ID, assume_unique=False, return_indices=True)

# Keep only Items Needed    
np_listA_Var1 = np.asarray( listA_Var1 )
np_listB_Var1 = np.asarray( listB_Var1 )

# Covert Array to List again
listA_ID=listB_ID=np_list_ID.tolist()
listA_Var1 = np_listA_Var1[listA_ind].tolist()
listB_Var1 = np_listB_Var1[listB_ind].tolist()