python - 将整数字符串列表转换为整数数组的最有效方法

Question

我有一个简单的问题 - 我需要将一个整数字符串转换为一个整数列表并将其插入到一个 numpy 数组中。

我有有效的代码，但我对更有效的方法感兴趣(如果有的话)。起始条件是我有一个整数字符串列表(第 4 行)，目标是获得一个用这些整数填充的 numpy 数组。

这是我使用的代码示例:

import numpy as np
print("Hello StackOverflow")

listOfStringOfINTs = ["123231231231231"]*5
print(listOfStringOfINTs)
numpyVectorOfInts = np.empty([len(listOfStringOfINTs),len(listOfStringOfINTs[0]) ], dtype='int')
for i, IntString in enumerate(listOfStringOfINTs):
    numpyVectorOfInts[i] = list(map(int, IntString))

print(numpyVectorOfInts)

Answer 1

我不确定这在速度上是否更好，但它更简单:

In [68]: np.array([list(astr) for astr in listOfStringOfINTs],int)           
Out[68]: 
array([[1, 2, 3, 2, 3, 1, 2, 3, 1, 2, 3, 1, 2, 3, 1],
       [1, 2, 3, 2, 3, 1, 2, 3, 1, 2, 3, 1, 2, 3, 1],
       [1, 2, 3, 2, 3, 1, 2, 3, 1, 2, 3, 1, 2, 3, 1],
       [1, 2, 3, 2, 3, 1, 2, 3, 1, 2, 3, 1, 2, 3, 1],
       [1, 2, 3, 2, 3, 1, 2, 3, 1, 2, 3, 1, 2, 3, 1]])

list(astr) 将字符串拆分为 1 个字符串的列表。具有 int dtype 的 np.array 负责转换所有字符串。

或者您可以将所有字符串连接成字符串，创建列表，然后 reshape 数组:

np.array(list(''.join(listOfStringOfINTs)),int).reshape(5,-1)