将 numpy 字符串数组转换为 numpy 矩阵

Question

所以我有一个 numpy 字符串数组，其中包含用空格分隔的数值，例如：

np.array(['1 2', '3 4'])
array(['1 2', '3 4'], dtype='<U3')

我想把它转换成一个数值矩阵，比如：

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

我正在寻找一种可以利用 numpy vecotrized 操作的操作，因为速度在这里很重要。在此示例中，行的长度为 2，但我需要一种具有任意行长度的通用方法。

谢谢！

Answer 1

如果它不必那么快，你可以逐个元素地迭代它然后应用：

def seperate_sting(s):

    split_numbers = s.split(' ')
    output = np.asarray(split_numbers).astype(int)

    return output


seperate_sting('1 1')
>>> array([1, 1])

Answer 2

首先，尝试用空格分割字符串，完成后检查函数numpy.asmatrix()

Answer 3

这是一种方法，假设非负整数成对出现，由一个空格隔开：

def to_num(x):                                          
    y = (x[:,None].view(np.int32)-48)*10**np.arange(x.itemsize//4-1,-1,-1)                    
    splt = y.argmin(1)                                                                        
    z = np.take_along_axis(y.cumsum(1),np.column_stack([splt-1,np.full(*y.shape-np.arange(2))]),1)
    z[:,1]+=10**(y.shape[1]-splt-1)*16-z[:,0]                                                    
    z[:,0]//=10**(y.shape[1]-splt)                                                               
    end = (y[:,::-1]>=0).argmax(1)
    z[:,1]+=np.concatenate([[0],48*np.cumsum(10**np.arange(end.max()))])[end]
    z[:,1]//=10**end
    return z

例如，10^6 对在我的机器上大约需要 3 秒：

from timeit import timeit

x = np.random.randint(0,1000000,(1000000,2))
x = np.array([" ".join(map(str, y)) for y in x.tolist()])

(to_num(x) == [[int(z) for z in y.split()] for y in x.tolist()]).all()
# True
timeit(lambda:to_num(x), number=10)
# 2.9360161621589214