python - 根据第一个删除第二个数组的索引位置

Question

我正在尝试用 python 对此进行编程。假设我有数组:

A = [0, 1, 1, 1, 1, 2, 2, 3]

B = ['A', 'A', 'A', 'E', 'E', 'D', 'D', 'C']

我想根据被丢弃的元素在A中的索引位置来丢弃数组B中的相应元素。例如，如果我在A中丢弃0:

A = [1, 1, 1, 1, 2, 2, 3]

然后 B 应该删除第一个 'A' 并变成:

B = ['A', 'A', 'E', 'E', 'D', 'D', 'C']

知道怎么做吗？感谢您的帮助。

Answer 1

在 python 中，有一些数组，例如在 numpy 中，但你指向的这些元素是列表，你可以使用 del 运算符删除这些元素，如果你想以自动方式执行此操作，你可以构建一个函数来计算它正确的，例如:

def removeFromBothLists(a, b, idx):
    del a[idx]
    del b[idx]

然后你可以调用它传递列表作为参数和你想删除的索引:

removeFromBothLists(a, b, 0)