要查找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()
函数将返回错误。