web-dev-qa-db-ja.com

numpyでの要素ごとの文字列連結

これはバグですか?

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

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

要素ごとに文字列を連結しようとしています。 Add()はnumpyでそれを行う方法であると思いましたが、明らかに期待どおりに機能していません。

29
Dave31415

これは 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データ型には、他にも便利な 文字列操作 があります。

46
Mike T

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')
10

numpyもPython文字列操作関数を内部的に使用するため、これは純粋なPythonで実行できます(また、実行する必要があります)。

>>> a1 = ['a','b']
>>> a2 = ['E','F']
>>> map(''.join, Zip(a1, a2))
['aE', 'bF']
6
Niklas B.

別の解決策は、文字列配列をオブジェクトの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)

これはそれほど遅くありません(整数配列を追加する場合の2倍未満です)。

1
jonathanrocher

もう1つの基本的でエレガントで高速なソリューション:

In [11]: np.array([x1 + x2 for x1,x2 in Zip(a1,a2)])
Out[11]: array(['aE', 'bF'], dtype='<U2')

小さいアレイでは非常に高速です。

In [12]: %timeit np.array([x1 + x2 for x1,x2 in Zip(a1,a2)])
3.67 µs ± 136 ns per loop (mean ± std. dev. of 7 runs, 100000 loops each)

In [13]: %timeit np.core.defchararray.add(a1, a2)
6.27 µs ± 28.3 ns per loop (mean ± std. dev. of 7 runs, 100000 loops each)

In [14]: %timeit np.char.array(a1) + np.char.array(a2)
22.1 µs ± 319 ns per loop (mean ± std. dev. of 7 runs, 10000 loops each)

より大きなアレイの場合、時間差はそれほど大きくありません。

In [15]: b1 = np.full(10000,'a')    
In [16]: b2 = np.full(10000,'b')    

In [189]: %timeit np.array([x1 + x2 for x1,x2 in Zip(b1,b2)])
6.74 ms ± 66.9 µs per loop (mean ± std. dev. of 7 runs, 100 loops each)

In [188]: %timeit np.core.defchararray.add(b1, b2)
7.03 ms ± 419 µs per loop (mean ± std. dev. of 7 runs, 100 loops each)

In [187]: %timeit np.char.array(b1) + np.char.array(b2)
6.97 ms ± 284 µs per loop (mean ± std. dev. of 7 runs, 100 loops each)
1
Gaurav Singhal