计算给定 NumPy 数组沿第二个轴的最大值和最小值之间的差

介绍：

本文介绍了如何使用NumPy计算给定数组沿第二个轴的最大值和最小值之间的差。

步骤：

1. 导入 NumPy 库

import numpy as np

2. 创建一个数组

arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

这个数组的形状是 (3, 3)，元素如下：

array([[1, 2, 3],
       [4, 5, 6],
       [7, 8, 9]])

3. 计算沿第二个轴的最大值和最小值之间的差

diff = np.ptp(arr, axis=1)

这个操作会返回一个形状为 (3,) 的数组，它的每个元素分别是第二个轴上对应行的最大值和最小值之间的差。

array([2, 2, 2])

4. 操作说明

• np.ptp 函数是 NumPy 中用于计算数组中数值范围的函数，它的用法如下：

np.ptp(a, axis=None, out=None, keepdims=<no value>)

• 参数 a 是需要计算的数组。
• 参数 axis 是需要计算的轴，缺省值为 None，表示计算整个数组。如果要计算特定的轴，可以指定轴的编号，例如 axis=1 表示计算二维数组的每行的数值范围。
• 参数 out 是可选的输出数组，如果指定了，则结果会赋值给该数组。
• 参数 keepdims 是可选的 Boolean 类型的参数，如果为 True，则结果数组保留原始数组的维度，否则结果数组会折叠为一个或多个轴上的一维数组。

参考资料：

• NumPy Documentation: numpy.ptp