如何为np.array元素及np.arange生成的id添加+号

我们默认将「添加+号」分为两种场景，你可以根据自身需求选择：

• 场景A：将数组元素转为带+号前缀的字符串类型，适合后续字符串拼接、导出等操作
• 场景B：仅在打印/输出时显示+号，数组本身保留数值类型，适合运算场景

需求1：为普通np.array的每一个元素添加+号 #

场景A实现（转带+号的字符串） #

import numpy as np

# 示例输入数组
arr = np.array([1, -2, 3, 0, -5])

# 方案1：向量化操作，大数组性能最优
arr_with_sign = np.where(arr >= 0, '+' + arr.astype(str), arr.astype(str))

# 方案2：列表推导，适合小数组快速编写
arr_with_sign = np.array([f"+{x}" if x >= 0 else str(x) for x in arr])

输出示例：array(['+1', '-2', '+3', '+0', '-5'], dtype='<U3')

场景B实现（仅输出显示+号，保留数值类型） #

import numpy as np

arr = np.array([1, -2, 3, 0, -5])
# 打印时自动添加+号，原数组不变
print(np.array2string(arr, sign='+'))

输出示例：[ +1 -2 +3 +0 -5]

需求2：为np.arange(len(web))生成的id统一添加+号 #

通过np.arange(len(web))生成的id默认都是从0开始的非负整数，无需判断正负，处理逻辑更简单：

场景A实现（转带+号的字符串） #

import numpy as np

# 示例web对象，可替换为你自己的可计算长度的对象
web = ["index.html", "about.html", "product.html"]

# 生成id并添加+号
ids = np.arange(len(web))
ids_with_sign = '+' + ids.astype(str)

输出示例：array(['+0', '+1', '+2'], dtype='<U3')

场景B实现（仅输出显示+号，保留数值类型） #

import numpy as np

web = ["index.html", "about.html", "product.html"]
ids = np.arange(len(web))
print(np.array2string(ids, sign='+'))

输出示例：[+0 +1 +2]

内容的提问来源于stack exchange，提问作者user458