Python中含字符串与数字的numpy数组相加报错及结果实现求助

问题原因 #

你创建的numpy数组因包含字符串'A'和整数，默认被设为object dtype。直接执行+操作时，遇到整数与字符串的元素对（比如第二个元素2和'5'），Python不支持整数与字符串直接相加，因此抛出TypeError。如果不指定dtype=object，numpy会自动把所有元素转为字符串，相加后会变成字符串拼接（比如'2'+'5'='25'），也不符合你的需求。

解决方案 #

方法1：列表推导式遍历处理（简单直观） #

通过遍历两个数组的元素对，判断是否为'A'，是则保留，否则执行整数加法，最后转回numpy数组：

import numpy as np

# 显式指定dtype=object，保留原始混合类型
array_a = np.array(['A', 2, 6, 5, 8, 9, 3, 6, 6, 0, 1], dtype=object)
array_b = np.array(['A', 5, 7, 1, 8, 9, 4, 6, 0, 5, 1], dtype=object)

# 遍历处理每个元素对
array_c = np.array([
    a if a == 'A' else a + b
    for a, b in zip(array_a, array_b)
], dtype=object)

print(array_c)
# 输出：array(['A', 7, 13, 6, 16, 18, 7, 12, 6, 5, 2], dtype=object)

方法2：numpy向量化操作（高效处理大数组） #

利用numpy的掩码和向量化函数，先处理数值部分相加，再还原字符串位置：

import numpy as np

array_a = np.array(['A', 2, 6, 5, 8, 9, 3, 6, 6, 0, 1], dtype=object)
array_b = np.array(['A', 5, 7, 1, 8, 9, 4, 6, 0, 5, 1], dtype=object)

# 创建掩码，标记所有为'A'的位置
mask = array_a == 'A'

# 将数组转为整数类型，把'A'临时替换为0（不影响加法结果）
num_a = np.where(mask, 0, array_a).astype(int)
num_b = np.where(mask, 0, array_b).astype(int)

# 数值相加后，把'A'的位置替换回来
array_c = np.where(mask, 'A', num_a + num_b).astype(object)

print(array_c)
# 输出：array(['A', 7, 13, 6, 16, 18, 7, 12, 6, 5, 2], dtype=object)

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