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

  • Post category:Python

要查找Numpy数组中每个字符串元素的长度,可以使用Numpy中的char模块中的len()函数。下面是详细的攻略步骤:

1. 创建包含字符串的Numpy数组

首先,需要创建一个包含字符串的Numpy数组。我们可以使用numpy.array()函数创建一个一维的字符串数组,具体代码如下:

import numpy as np
arr = np.array(['Hello', 'World', 'Numpy'])

这将创建一个包含三个字符串的Numpy数组。

2. 导入char模块

要使用char模块中的len()函数,需要先从Numpy中导入char模块。具体代码如下:

from numpy import char

3. 使用char模块中的len()函数

在导入char模块后,现在就可以使用len()函数查找每个字符串元素的长度了。具体代码如下:

str_length = char.str_len(arr)

这将创建一个数组str_length,其中包含每个字符串元素的长度。

示例

下面是使用以上攻略查找一个包含字符串的Numpy数组中每个字符串元素的长度的两个示例:

import numpy as np
from numpy import char

# 示例 1
arr1 = np.array(['Hello', 'World', 'Numpy'])
str_length1 = char.str_len(arr1)
print(str_length1)

# 示例 2
arr2 = np.array(['Python', 'Numpy', 'Pandas', 'Scipy'])
str_length2 = char.str_len(arr2)
print(str_length2)

输出结果分别为:

[5 5 5]
[6 5 6 5]

可以看到,这些示例中,我们全部成功地找到了每个字符串元素的长度。