跨NumPy数组矢量化字符串格式

Question

我有两个整数数组，每个元素要组合成一个'a[i]_b[i]'形式的字符串数组。也就是说，我有

import numpy as np
a = np.array([1, 2, 3])
b = np.array([4, 5, 6])

我想得到数组

result = np.array(['1_4', '2_5', '3_6'])

我可以很容易地做到这一点与列表理解：

result = np.array(['{}_{}'.format(a[i], b[i]) for i in range(len(a))])

但是我的数组a和b非常长，这太慢了。有没有办法通过numpy电话来做到这一点？什么是hstack式的？我能做到这一点：

result np.hstack([a.resahpe(len(a), 1), b.reshape(len(b), 1)])

这让我

np.array([[1, 4]
          [2, 5]
          [3, 6]])

当然，一定有一个简单的方法从这里得到我想要的结果。。。你知道吗

编辑：

使用两个defchararray调用似乎比列表理解更慢，但是frompyfunc是有效的。。。你知道吗

>>> import timeit
>>> t = Timer("""['{}_{}'.format(a[i], b[i]) for i in range(len(b))]""", setup="""import numpy as np; a = np.random.randn(10000); b=np.random.randn(10000)""")
>>> t.timeit(1000)
22.310123541974463

>>> t2 = Timer("""np.core.defchararray.add(np.core.defchararray.add(a.astype(str),'_'), b.astype(str))""", setup="""import numpy as np; a = np.random.randn(10000); b=np.random.randn(10000)""")
>>> t2.timeit(1000)
28.489826270961203

>>> t3 = Timer("""fv(a,b)""", setup='import numpy as np; a = np.random.randn(10000); b=np.random.randn(10000); fv =  np.frompyfunc("{}_{}".format, 2, 1)')
>>> t3.timeit(1000)
15.455791965010576

这里它是a和b长度的函数（看起来很稳定）：

Answer 1

fv = np.frompyfunc("{}_{}".format, 2, 1)
result = fv(a, b)  # array(['1_4', '2_5', '3_6'], dtype=object)

根据一些时间安排，这大约是列表理解和转换的运行时间的1/3np.数组. 你知道吗

In [2]: a = np.arange(100000)

In [3]: b = np.arange(100000) + a.size

In [4]: fv = np.frompyfunc("{}_{}".format, 2, 1)

In [5]: def f(a, b): return np.array(["{}_{}".format(a,b) for a,b in zip(a,b)], dtype=object)

In [6]: %timeit f(a,b)
370 ms ± 12.5 ms per loop (mean ± std. dev. of 7 runs, 1 loop each)

In [7]: %timeit fv(a,b)
137 ms ± 1.48 ms per loop (mean ± std. dev. of 7 runs, 10 loops each)

Answer 2

你可以使用defchararray.add

import numpy as np
a = np.array([1, 2, 3])
b = np.array([4, 5, 6])
c = np.core.defchararray.add(a.astype(str),'_')
c = np.core.defchararray.add(c,b.astype(str))
print(c) #['1_4' '2_5' '3_6']