查找Numpy数组中每个字符串元素的长度

  • Post category:Python

要查找NumPy数组中每个字符串元素的长度,可以使用NumPy中的np.char模块中的len()函数。

首先,需要导入NumPy库:

import numpy as np

然后,定义一个包含字符串的NumPy数组:

arr = np.array(['hello', 'world', 'numpy'])

可以使用np.char.len()函数来查找每个字符串元素的长度,例如:

lengths = np.char.len(arr)
print(lengths)

输出结果为:

[5 5 5]

另外一个例子是对多维数组进行操作。定义一个包含多个字符串的NumPy数组:

arr = np.array([['hello', 'world'], ['numpy', 'array']])

可以使用np.char.len()函数来查找每个字符串元素的长度,例如:

lengths = np.char.len(arr)
print(lengths)

输出结果为:

[[5 5]
[5 5]]

需要注意的是,对于非字符串类型的元素,np.char.len()函数将返回错误。