NumPyで配列の形状を変更する方法

NumPyのreshape()関数を使用すると、配列の形状を変更できます。以下にreshape()関数を使用した形状変更の例を示します。

import numpy as np

# 创建一个一维数组
arr = np.array([1, 2, 3, 4, 5, 6])

# 将一维数组转换为二维数组
arr_2d = arr.reshape(2, 3)
print(arr_2d)
# 输出:
# [[1 2 3]
#  [4 5 6]]

# 创建一个二维数组
arr_2d = np.array([[1, 2, 3], [4, 5, 6]])

# 将二维数组转换为一维数组
arr_1d = arr_2d.reshape(6)
print(arr_1d)
# 输出:
# [1 2 3 4 5 6]

# 可以使用-1来自动推断数组的形状
arr_1d = arr_2d.reshape(-1)
print(arr_1d)
# 输出:
# [1 2 3 4 5 6]

# 创建一个三维数组
arr_3d = np.array([[[1, 2], [3, 4]], [[5, 6], [7, 8]]])

# 将三维数组转换为二维数组
arr_2d = arr_3d.reshape(4, 2)
print(arr_2d)
# 输出:
# [[1 2]
#  [3 4]
#  [5 6]
#  [7 8]]

reshape()関数を用いて配列の形状を変更する際、新しい形状の総要素数が元の配列の総要素数と同じであることを確認してください。そうでなければ、ValueError例外が発生します。