python - 添加到 numpy 字符串数组

Question

我有以下两个 numpy 数组:

a1
array(['T', 'T', 'T', ..., 'G', 'A', 'A'], dtype=object)

a2
array(['A', 'G', 'C', ..., 'A', 'T', 'G'], dtype=object)

我想添加两个数组中的每个元素。以下命令有效:

s = a1 + a2
s
array(['TA', 'TG', 'TC', ..., 'GA', 'AT', 'AG'], dtype=object)

但是我在尝试使用 numpy.char.add 时遇到错误:

np.char.add(a1, a2)

TypeError: string operation on non-string array

为什么会这样？

Answer 1

你的类型是 object 尝试将它设置为 str 它会起作用:

import numpy as np

a1=np.array(['T', 'T', 'T', ..., 'G', 'A', 'A'], dtype=str)

a2=np.array(['A', 'G', 'C', ..., 'A', 'T', 'G'], dtype=str)

print(np.char.add(a1, a2))