Python – 使用shape函數進行數組維度操作

一、背景介紹

在Python中，數組（或者矩陣）是一種由相同類型的元素組成的集合。在數據處理和科學計算中，處理和操作多維數組是很常見的需求。Python提供了許多函數，用於創建和操作數組。在這些函數中，shape函數是一個常用的函數，它可以用來獲取或修改數組的維度。

二、使用shape函數獲取數組的維度信息

我們可以使用shape函數來獲取一個數組的維度信息。下面的代碼演示了如何使用shape函數獲取一個二維數組的維度：

import numpy as np

# 創建一個二維數組
a = np.array([[1, 2], [3, 4], [5, 6]])

# 獲取a數組的維度信息，輸出結果為(3, 2)
print(a.shape)

上面的代碼中，我們使用NumPy模塊創建了一個二維數組a。然後，我們使用a.shape獲取了a數組的維度信息。

三、使用shape函數修改數組的維度

除了獲取數組的維度信息，我們也可以使用shape函數來修改數組的維度。下面的代碼演示了如何使用shape函數修改一個二維數組的維度：

import numpy as np

# 創建一個二維數組
a = np.array([[1, 2], [3, 4], [5, 6]])

# 使用shape函數將a數組的維度修改為(2, 3)
a.shape = (2, 3)

# 輸出修改後的a數組
print(a)

上面的代碼中，我們使用NumPy模塊創建了一個二維數組a。然後，我們使用a.shape修改了a數組的維度為(2, 3)。最後，我們輸出修改後的a數組。

四、使用resize函數修改數組的維度

除了使用shape函數修改數組的維度外，還可以使用resize函數來修改數組的維度。和shape函數不同的是，resize函數能夠自動填充或切割數組的元素。下面的代碼演示了如何使用resize函數修改一個二維數組的維度：

import numpy as np

# 創建一個二維數組
a = np.array([[1, 2], [3, 4], [5, 6]])

# 使用resize函數將a數組的維度修改為(2, 3)，並自動填充或切割元素
a.resize(2, 3)

# 輸出修改後的a數組
print(a)

上面的代碼中，我們使用NumPy模塊創建了一個二維數組a。然後，我們使用a.resize函數修改了a數組的維度為(2, 3)。由於原數組a的大小和新大小不匹配，因此數組自動填充或切割元素。最後，我們輸出修改後的a數組。

五、使用reshape函數修改數組的維度

除了使用shape函數和resize函數修改數組的維度外，還可以使用reshape函數來修改數組的維度。和resize函數類似，reshape函數也能夠自動填充或切割數組的元素。下面的代碼演示了如何使用reshape函數修改一個二維數組的維度：

import numpy as np

# 創建一個二維數組
a = np.array([[1, 2], [3, 4], [5, 6]])

# 使用reshape函數將a數組的維度修改為(2, 3)
a = a.reshape(2, 3)

# 輸出修改後的a數組
print(a)

上面的代碼中，我們使用NumPy模塊創建了一個二維數組a。然後，我們使用a.reshape函數修改了a數組的維度為(2, 3)。由於原數組a的大小和新大小不匹配，因此數組自動填充或切割元素。最後，我們輸出修改後的a數組。

六、總結

通過本文的介紹，我們學習了如何使用shape函數獲取和修改數組的維度，以及如何使用resize函數和reshape函數修改數組的維度。這些函數對於進行數組維度操作是非常有幫助的。