要将一个 6 × 5 6 \times 5 6×5 的数组变换为三个新的 2 × 5 2 \times 5 2×5 数组,并且让原数组的第1行和第4行成为第一个新数组,第2行和第5行成为第二个新数组,第3行和第6行成为第三个新数组,可以使用NumPy的索引和切片功能。
以下是实现这一转换的代码示例:
import numpy as np
# 创建一个 6x5 的示例数组
original_array = np.arange(30).reshape(6, 5)
# 按要求进行重组
new_array1 = original_array[[0, 3], :]
new_array2 = original_array[[1, 4], :]
new_array3 = original_array[[2, 5], :]
# 显示结果
print("Original Array:\n", original_array)
print