以下是一个示例代码,演示了如何保持多个列相同,仅调整两列的形状:
import numpy as np
# 创建一个具有5个行和3个列的示例数组
arr = np.array([[1, 2, 3],
[4, 5, 6],
[7, 8, 9],
[10, 11, 12],
[13, 14, 15]])
# 打印原始数组
print("原始数组:")
print(arr)
# 选择要调整形状的两列
col1 = arr[:, 1]
col2 = arr[:, 2]
# 创建一个新的形状为(5, 2)的数组,其中两列分别为col1和col2的值
new_arr = np.column_stack((col1, col2))
# 打印调整形状后的数组
print("调整形状后的数组:")
print(new_arr)
输出结果:
原始数组:
[[ 1 2 3]
[ 4 5 6]
[ 7 8 9]
[10 11 12]
[13 14 15]]
调整形状后的数组:
[[ 2 3]
[ 5 6]
[ 8 9]
[11 12]
[14 15]]
在上面的代码中,我们首先创建了一个形状为(5, 3)的示例数组。然后,我们选择了数组的第二列和第三列,并将它们保存在变量col1和col2中。接下来,我们使用np.column_stack()函数将这两列组合成一个新的形状为(5, 2)的数组new_arr。最后,我们打印出调整形状后的数组。
上一篇:保持多个会话
下一篇:保持多个命名管道打开是不好的吗?