用于删除数字中可变长度序列的正则表达式匹配

Question

我有一个看起来像 X000Y000 的 numpy 整数数组。 X 和 Y 的长度可以是 1、2 或 3，并且可能包含 0。我想将数组中的每个元素都转换为 X。我觉得可以为此使用正则表达式，但无法找到一个好的使用方法，或者如何将其应用于整个数组。

例如：14000010000 应该变成 140。

Answer 1

我假设X 和Y 不能以0 开头。 [1-9]\d{0,2} 匹配不以 0 开头的 1 到 3 位数字。

所以提取X和Y的正则表达式应该是：

^([1-9]\d{0,2})000([1-9]\d{0,2})000$

然后您可以使用re.sub() 删除X 和Y 之间的零。

regex = re.compile(r'^([1-9]\d{0,2})000([1-9]\d{0,2})000$');
i = 14000010000
istr = str(i)
new_i = int(regex.sub(r'\1\2', istr)

你可以把它映射到你的 numpy 数组上

regex = re.compile(r'^([1-9]\d{0,2})000([1-9]\d{0,2})000$');
new_array = np.fromiter((int(regex.sub(r'\1\2', str(x)) for x in array), array.dtype)

请参阅Most efficient way to map function over numpy array，了解将函数映射到 numpy 数组的各种方法。