要过滤出给定特定字母的ndarray,可以使用NumPy库提供的功能来实现。具体步骤如下:
import numpy as np
array()
函数创建一个ndarray。arr = np.array(['apple', 'banana', 'cat', 'dog', 'elephant'])
find()
方法或NumPy的where()
函数来实现。使用字符串的find()
方法:
letter = 'a'
filtered_arr = np.array([word for word in arr if word.find(letter) != -1])
使用NumPy的where()
函数:
letter = 'a'
filtered_arr = arr[np.where([word.find(letter) != -1 for word in arr])]
print(filtered_arr)
完整示例代码如下:
import numpy as np
arr = np.array(['apple', 'banana', 'cat', 'dog', 'elephant'])
letter = 'a'
filtered_arr = arr[np.where([word.find(letter) != -1 for word in arr])]
print(filtered_arr)
以上代码将输出包含字母'a'的元素,结果为:
['apple' 'banana' 'cat']
这是一个简单的例子,但可以根据具体需求进行扩展和修改。
没有搜到相关的沙龙
领取专属 10元无门槛券
手把手带您无忧上云