python - 将列表的列表转换为 numpy 数组时保持原始数据类型

Question

将列表的列表转换为numpy数组时，如何保持原始数据类型？

我使用 np.array、np.matrix 将列表的列表转换为 numpy 数组。但是结果发现所有的int都变成了string。 Python 版本为 3.7.x。

X = [[3, 'aa', 10],                 
     [1, 'bb', 22],                      
     [2, 'cc', 28],                      
     [5, 'bb', 32],                      
     [4, 'cc', 32]]
# X is a list of list
X = np.array(X)
return X

# X becomes
[['3' 'aa' '10']
 ['1' 'bb' '22']
 ['2' 'cc' '28']
 ['5' 'bb' '32']
 ['4' 'cc' '32']]

Answer 1

改用 X = np.array(X, dtype="O")。然后每个项目都存储为 Python 对象。