python - 从数组中删除元素，相应地更新存储索引列表

Question

考虑以下形式的 numpy 数组:

> a = np.random.uniform(0., 100., (10, 1000))

以及我想要跟踪的数组中元素的索引列表:

> idx_s = [0, 5, 7, 9, 12, 17, 19, 32, 33, 35, 36, 39, 40, 41, 42, 45, 47, 51, 53, 57, 59, 60, 61, 62, 63, 65, 66, 70, 71, 73, 75, 81, 83, 85, 87, 88, 89, 90, 91, 93, 94, 96, 98, 100, 106, 107, 108, 118, 119, 121, 124, 126, 127, 128, 129, 133, 135, 138, 142, 143, 144, 146, 147, 150]

我还有一个需要从 a 中删除的元素索引列表:

> idx_d = [4, 12, 18, 20, 21, 22, 26, 28, 29, 31, 37, 43, 48, 54, 58, 74, 80, 86, 99, 109, 110, 113, 117, 134, 139, 140, 141, 148, 154, 156, 160, 166, 169, 175, 183, 194, 198, 199, 219, 220, 237, 239, 241, 250]

我删除它:

> a_d = np.delete(arr, idx_d, axis=1)

但是这个过程会改变a_d中元素的索引。 idx_s 中的索引不再指向 a_d 中 a 中的相同元素，因为 np.delete()感动了他们。例如:如果我从a中删除索引4的元素，那么idx_s中4之后的所有索引都是现在在 a_d 中向右移动了 1。

                 v Index 5 points to 'f' in a
       0 1 2 3 4 5 6
a   -> a b c d e f g ... # Remove 4th element 'e' from a
a_d -> a b c d f g h ... # Now index 5 no longer points to 'f' in a_d, but to 'g'
       0 1 2 3 4 5 6

如何更新索引的 idx_s 列表，以便 a 中指向的相同元素也指向 a_d 中？

如果idx_s中存在的元素也存在在idx_d中(因此从a中删除) 并且不存在于 a_d 中)，其索引也应被丢弃。

Answer 1

您可以使用np.searchsorted获取 idx_s 中每个元素的移位，然后只需从 idx_s 中减去这些值即可得到新的下移值，如下 -

idx_s - np.searchsorted(idx_d, idx_s)

如果idx_d尚未排序，我们需要提供排序版本。因此，为了简单起见，假设它们是数组，我们将 -

idx_s = idx_s[~np.in1d(idx_s, idx_d)]
out = idx_s - np.searchsorted(np.sort(idx_d), idx_s)

帮助获得更好图片的示例运行 -

In [530]: idx_s
Out[530]: array([19,  5, 17,  9, 12,  7,  0])

In [531]: idx_d
Out[531]: array([12,  4, 18])

In [532]: idx_s = idx_s[~np.in1d(idx_s, idx_d)] # Remove matching ones

In [533]: idx_s
Out[533]: array([19,  5, 17,  9,  7,  0])

In [534]: idx_s - np.searchsorted(np.sort(idx_d), idx_s) # Updated idx_s
Out[534]: array([16,  4, 15,  8,  6,  0])