numpy中的元素字符串连接

Question

这是一个错误吗？

import numpy as np
a1=np.array(['a','b'])
a2=np.array(['E','F'])

In [20]: add(a1,a2)
Out[20]: NotImplemented

我正在尝试按元素进行字符串连接.我认为Add()是在numpy中实现它的方法,但显然它没有按预期工作.

Answer 1

这可以使用numpy.core.defchararray.add完成.这是一个例子:

>>> import numpy as np
>>> a1 = np.array(['a', 'b'])
>>> a2 = np.array(['E', 'F'])
>>> np.core.defchararray.add(a1, a2)
array(['aE', 'bF'], 
      dtype='<U2')

NumPy数据类型还有其他有用的字符串操作.

Answer 2

您可以使用chararray子类对字符串执行数组操作:

a1 = np.char.array(['a', 'b'])
a2 = np.char.array(['E', 'F'])

a1 + a2
#chararray(['aE', 'bF'], dtype='|S2')

另一个好例子:

b = np.array([2, 4])
a1*b
#chararray(['aa', 'bbbb'], dtype='|S4')

Answer 3

这可以(并且应该)在纯Python中完成,因为numpy内部也使用Python字符串操作函数:

>>> a1 = ['a','b']
>>> a2 = ['E','F']
>>> map(''.join, zip(a1, a2))
['aE', 'bF']

Answer 4

另一个解决方案是将字符串数组转换为 python 对象数组，以便调用 str.add ：

>>> import numpy as np
>>> a = np.array(['a', 'b', 'c', 'd'], dtype=np.object)   
>>> print a+a
array(['aa', 'bb', 'cc', 'dd'], dtype=object)

这并没有那么慢（比添加整数数组慢不到两倍）。