熱線電話:13121318867

登錄
首頁大數據時代請問numpy中怎么刪掉值為1的維度?
請問numpy中怎么刪掉值為1的維度?
2023-04-28
收藏

在 NumPy 中,可以使用 np.squeeze() 函數來刪除值為 1 的維度。本文將詳細介紹 np.squeeze() 函數的用法和示例。

什么是 np.squeeze() 函數?

np.squeeze() 函數是 NumPy 庫中的一個函數,用于從數組的形狀中刪除單維條目。如果將一個數組作為參數傳遞給 np.squeeze(),則該函數將返回一個新數組,其中所有維度為 1 的軸都被刪除。此外,如果您希望只刪除特定的維度,請在 np.squeeze() 函數的第二個參數中指定要刪除的維度。

np.squeeze() 函數的語法

np.squeeze() 函數的語法如下所示:

numpy.squeeze(arr, axis=None)

其中:

  • arr:需要壓縮的數組。
  • axis:默認為 None,表示刪除所有維度為 1 的軸,也可以指定要刪除的特定軸。

np.squeeze() 函數的示例

現在,讓我們通過一些示例來了解 np.squeeze() 函數的使用。

示例 1:刪除所有維度為 1 的軸

假設有一個形狀為 (1, 2, 1, 3) 的數組。使用 np.squeeze() 函數可以從數組中刪除維度為 1 的軸,并返回形狀為 (2, 3) 的新數組。

import numpy as np

a = np.ones((1, 2, 1, 3))
b = np.squeeze(a)

print("Shape of a:", a.shape)
print("Shape of b:", b.shape)

輸出:

Shape of a: (1, 2, 1, 3)
Shape of b: (2, 3)

示例 2:刪除指定的維度

在這個例子中,我們將僅刪除第二維和第四維。

import numpy as np

a = np.ones((1, 2, 3, 1, 4))
b = np.squeeze(a, axis=(1, 3))

print("Shape of a:", a.shape)
print("Shape of b:", b.shape)

輸出:

Shape of a: (1, 2, 3, 1, 4)
Shape of b: (1, 3, 4)

總結

np.squeeze() 函數可以方便地刪除數組中單維條目。您可以使用它來刪除特定軸上的值為 1 的維度,也可以使用它來刪除所有維度為 1 的軸。希望本文能夠對您理解 np.squeeze() 函數有所幫助。

數據分析咨詢請掃描二維碼

若不方便掃碼,搜微信號:CDAshujufenxi

數據分析師資訊
更多

OK
客服在線
立即咨詢
日韩人妻系列无码专区视频,先锋高清无码,无码免费视欧非,国精产品一区一区三区无码
客服在線
立即咨詢