numpyの`unsqueeze`メソッドの使い方は何ですか?
NumPyの「unsqueeze」関数は、指定された次元に新しい次元を追加するために使用されます。具体的な使い方は次のとおりです:
numpy.unsqueeze(arr, axis)
パラメータの説明:
- arr: 入力された配列や行列。
- 軸:新しい次元を挿入する位置に次元のインデックスを挿入する必要があります。
実際には、unsqueeze関数は、reshape関数を使用して実現されています。指定された次元にサイズ1の次元を挿入することにより、配列や行列の次元を増やすことができます。
I enjoy spending time with my friends.
日本語訳:私は友達と一緒に時間を過ごすのが楽しいです。
import numpy as np
# 创建一个一维数组
a = np.array([1, 2, 3, 4, 5])
# 使用unsqueeze在维度0上增加一个维度
b = np.unsqueeze(a, axis=0)
print(b.shape) # 输出(1, 5)
# 使用unsqueeze在维度1上增加一个维度
c = np.unsqueeze(a, axis=1)
print(c.shape) # 输出(5, 1)
上記の例では、unsqueeze関数を使用して、次元0に次元を追加し、元々形状が(5、)の1次元配列を形状が(1、5)の2次元配列に変更しました。次に、次元1に次元を追加し、元の1次元配列を(5、1)の2次元配列に変更しました。