重新编号一维数组的元素

Question

我是新来的，我想和你分享一个我面临的问题

基本上我必须根据另一个数组的元素重新编号一个数组。

更多细节：

我有一个 int 数组 a

a=array([    1,     2,     3, ..., 21041, 21042, 21043])

和

len(a)=21043

然后，将掩码应用于“a”，得到 b

b=a[mask]
b=array[1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 28 32 33 34 47 54 55 62....21043]

和

len(b)=15717

在这一点上，我已经轻松地对这个数组进行了排序：

b_renumber=array[1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23....15717]

因此，“b”中的数字 28 在“b_renumber”中变为 21，32 变为 22，依此类推。

现在问题是第三个数组 c

因为 c 的长度大于 b。

len(c)=46153

并且由下式给出：

c=array[ 3 4 142 633 12 19 564 497 513 54 308 177 254 532 155 3 273 28 ...21037]

所以基本上 c 是一个长度为 46153（未排序）的数组，其中 go 从 1 到 21043 的元素，以及数组 b 和像 b 一样的一些数字不存在（例如 21 22 23... b 中不存在的相同元素也在 c 中不存在。

我想要做的是重新编号 c 的元素，这样 c 的元素值和 b_renumber 之间存在链接。 b中的示例我们有数字28，在b_renumber中变成21（从b到b_renumber，28 -> 21），我想做的是：每次我在c中遇到值28时，这必须变成21，其他数字相同。

最后一点：数组仍然很小，但我必须使用更大维度的数组，所以我必须以有效的方式执行此操作，可能存在 function 或一些 package 允许我这样做。 请我愿意接受每一个建议

很抱歉帖子的长度，我希望有人可以帮助我。

先感谢您。

马泰奥

Answer 1

假设 b_renum 只包含 b 中每个元素的索引+1：

输入：

b = [0,1,2,100]
c = [20,50,60,100]

请注意dict.get(key, default)如果匹配则返回键的值，否则返回默认值。

b_renum = list(range(1,len(b)+1))
renum_dict = dict(zip(b, b_renum))
# slightly slower
# renum_dict = {elem:ind for ind,elem in enumerate(b)}

c_renum = [renum_dict.get(key,key) for key in c]

>>> print(b_renum)
[1, 2, 3, 4]

>>> print(c_renum)
[20, 50, 60, 3]

请注意 100 如何在c_renum中映射到 3。 我认为对您有用的是 1) 使用zip()因为您知道b和b_renum的长度相同，以及 2) 使用字典来查找值 + dict.get()技巧。 列表理解也很快。