初始化字符串数据的numpy数组的怪异行为

Question

当数组包含字符串数据时，我在使用numpy时遇到一些琐碎的麻烦。我有以下代码：

my_array = numpy.empty([1, 2], dtype = str)
my_array[0, 0] = "Cat"
my_array[0, 1] = "Apple"

现在，当我使用打印时print my_array[0, :]，得到的响应是['C', 'A']，显然不是Cat和Apple的预期输出。为什么会这样，如何获得正确的输出？

谢谢！

Answer 1

Numpy要求字符串数组具有固定的最大长度。当您使用创建空数组时dtype=str，默认情况下会将最大长度设置为1。你看你是否做my_array.dtype;
它将显示“ | S1”，表示“一个字符的字符串”。随后的数组分配将被截断以适应此结构。

您可以通过以下方式传递最大长度的显式数据类型，例如：

my_array = numpy.empty([1, 2], dtype="S10")

“ S10”将创建一个长度为10的字符串数组。您必须确定足够大的大小才能容纳要保留的所有数据。