如何在NumPy 1d-array中找到最大值和最小值

  • Post category:Python

在NumPy中,可以使用numpy.ndarray类代表一维数组,可以使用以下两个函数找到一个数组中的最大值和最小值:

  • numpy.amax(a[, axis, out, keepdims, initial, where]):返回数组a中的最大值。
  • numpy.amin(a[, axis, out, keepdims, initial, where]):返回数组a中的最小值。

其中,参数a为要计算最大/最小值的NumPy数组。可以通过设置axis参数来沿着指定轴计算最大/最小值,keepdims参数可以控制计算后结果的维度是否保留(True代表保留,False代表不保留,默认为False)。

以下是两条在NumPy 1d-array中找到最大值和最小值的示例:

示例一:

import numpy as np

a = np.array([10, 20, 30, 40])
max_value = np.amax(a)
min_value = np.amin(a)

print('max_value:', max_value)
print('min_value:', min_value)

输出结果:

max_value: 40
min_value: 10

在上面的示例中,我们首先导入NumPy库并创建了一个包含四个元素的一维数组。接着,通过numpy.amax()numpy.amin()函数来计算数组a中的最大值和最小值,最后打印出计算结果。

示例二:

import numpy as np

a = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
max_value = np.amax(a, axis=0)
min_value = np.amin(a, axis=1)

print('max_value:', max_value)
print('min_value:', min_value)

输出结果:

max_value: [7 8 9]
min_value: [1 4 7]

在上面的示例中,我们创建了一个包含3×3个元素的二维数组a,并使用numpy.amax()numpy.amin()函数来计算数组a中每列的最大值和每行的最小值。在计算数组a中每列的最大值时,设置了axis=0,表示沿着第0维计算;在计算数组a中每行的最小值时,设置了axis=1,表示沿着第1维计算。最后打印出了计算结果。