用分隔符python连接字符矩阵

Posted 2021-03-29

我正在尝试创建一个大型的numpy字符数组（5+维度），其中每个bin表示数字的组合。这些数字我将消除（可能带/）所以我可以在需要时将字符串拆分。因此，特别是当它变得像20维阵列时，感兴趣的是节省时间。我希望用矩阵来做这个，而不是有20个循环。请在下面找到一个带循环的二维示例，以及如何在循环中执行此操作的任何想法都会有所帮助。

import numpy as np
storing_array = np.chararray( [80, 98], itemsize = 100 )
for x1 in range(0, 80):
  for x2 in range(0, 98):
    storing_array[x1, x2] = '/'.join([str(x1), str(x2)])

谢谢！

答案

您可以使用x1方便地定义您的x2和np.arange，将它们转换为str然后将它们转换为object，以便它们支持某些操作：

x2=np.arange(98).astype(str).astype(object)
x1=np.arange(80).astype(str).astype(object)

现在将/添加到x1：

x1+='/'

现在使用np.broadcasting获得所有组合：

result=x1[:,None]+x2