在替换numpy数组中的值时防止字符串被截断

通过设置 dtype 属于 b "object"

import numpy as np
a = np.array([1,2,3])
b = np.array(['red','red','red'], dtype="object")

b[a<3] = "blue"

print(b)

这将输出:

['blue' 'blue' 'red']

数据类型 将处理字符串或其他常规Python对象。这也必然意味着在引擎盖下你会有一个 numpy

import numpy as np

a = np.array([1,2,3])
b = np.array(['red','red','red'])

replacement = "blue"
b = b.astype('<U{}'.format(max(len(replacement), a.dtype.itemsize)))
b[a<3] = replacement
print(b)

replacement 比数组中所有现有字符串都长。

如果构造这样的数组,则类型如下所示:

>>> b
array(['red', 'red', 'red'], dtype='<U3')

这意味着字符串的长度最多为3个字符。如果指定较长的字符串,则这些字符串 截断的 .

您可以更改数据类型以使最大长度更长,例如:

b2 = b.astype('<U10')