如何获得一个给定的NumPy数组沿第二轴的最小值和最大值
让我们看看如何沿着第二轴获得一个给定的NumPy数组的最小值和最大值。在这里,第二轴指的是行的方向。
我们使用NumPy的numpy.max()和numpy.amin()函数来分别获得数组沿第二轴的最小和最大值。
numpy.max():该函数返回数组的最大值或沿轴的最大值(如果提到)。
语法: numpy.amax(a, axis=None, out=None, keepdims=<no value>, initial=<no value>, where=<no value>)
numpy.amin():该函数返回数组的最小值或沿轴的最小值(如果提到的话)。
语法: numpy.amin(a, axis=None, out=None, keepdims=<no value>, initial=<no value>, where=<no value>)
现在,让我们看看一个例子。
示例 1:
# Import numpy library
import numpy as np
# Create a numpy array
arr = np.array([[0, 1],
[2, 3]])
print("Given Array:\n",
arr)
# find row-wise max values
rslt1 = np.amax(arr, 1)
print("\nMaximum Value:",
rslt1)
# find row-wise min values
rslt2 = np.amin(arr, 1)
print("\nMinimum Value:",
rslt2)
输出:
Given Array:
[[0 1]
[2 3]]
Maximum Value: [1 3]
Minimum Value: [0 2]
示例 2:
# Import numpy library
import numpy as np
# Create a numpy array
arr = np.array([[10, 34, 45],
[22, -3, 46],
[33, 4, 6]])
print("Given array:\n",
arr)
# find row-wise max values
rslt1 = np.amax(arr, 1)
print("\nMaximum value along the second axis:",
rslt1)
# find row-wise min values
rslt2 = np.amin(arr, 1)
print("\nMinimum value along the second axis:",
rslt2)
输出:
Given array:
[[10 34 45]
[22 -3 46]
[33 4 6]]
Maximum value: [45 46 33]
Minimum value: [10 -3 4]