如何利用 Numpy 打印给定范围内的数组元素?
在这个程序中,我们必须打印给定范围内的 numpy 数组元素。使用的不同的 numpy 函数为 numpy.where() 和 numpy.logical_and()。
算法
Step 1: Define a numpy array. Step 2: Use np.where() and np.logical_and() to find the numbers within the given range. Step 3: Print the result.
示例代码
import numpy as np arr = np.array([1,3,5,7,10,2,4,6,8,10,36]) print("Original Array:\n",arr) result = np.where(np.logical_and(arr>=4, arr<=20)) print(result)
输出
Original Array: [ 1 3 5 7 10 2 4 6 8 10 36] (array([2, 3, 4, 6, 7, 8, 9], dtype=int64),)
说明
结果给出了满足 np.where() 函数中条件的元素的索引位置。
广告