Python：如何使用NumPy ndarray创建字符串数组

Question

我正在尝试使用scikit learning使用K最近邻分类进行手写数字识别。我正在创建自己的数据集，并具有标签列表（其中标签为字符串）。列表的长度为5000。

但是，要进行KNN分类，图像数据和标签都必须位于numpy ndarrays中。 如何将字符串列表转换为包含这些字符串的NumPy ndarray？任何见解都会受到赞赏。

Answer 1

只需使用np.array()函数，如下所示：

import numpy as np

string_list = ["hello", "world", "str3", "str4"]

string_ndarray = np.array(string_list)

print(string_ndarray)
print(type(string_ndarray))

输出：

['hello' 'world' 'str3' 'str4']
<class 'numpy.ndarray'>