np.r_ 是 NumPy 库中的一个快捷方式,用于沿水平方向堆叠数组。它的功能类似于 np.concatenate 或 np.hstack。np.r_ 的主要优点是它的简洁性和便捷性,尤其是在交互式环境中。
np.r_ 的用法很简单,只需将要堆叠的数组作为参数传递给它。下面是一个例子:
import numpy as np
# 创建两个 1D 数组
array1 = np.array([1, 2, 3])
array2 = np.array([4, 5, 6])
# 使用 np.r_ 沿水平方向堆叠数组
result = np.r_[array1, array2]
# 输出结果
print(result) # 结果为 [1, 2, 3, 4, 5, 6],两个数组已沿水平方向堆叠
np.r_ 也支持多维数组的堆叠。例如,要沿第二个维度(列)堆叠两个二维数组,可以使用以下代码:
import numpy as np
# 创建两个 2x2 的二维数组
array1 = np.array([[1, 2],
[3, 4]])
array2 = np.array([[5, 6],
[7, 8]])
# 使用 np.r_ 沿第二个维度(列)堆叠数组
result = np.r_['1', array1, array2]
# 输出结果
print(result)
# 结果为:
# [[1, 2, 5, 6],
# [3, 4, 7, 8]]
在这个例子中,‘1’ 是一个附加参数,用于指定应沿哪个轴堆叠数组。这里我们指定了沿第二个维度(列)堆叠数组。