使用numpy數組添加元素

引言

在 Python 中， Numpy 是一個非常受歡迎的科學計算庫，主要用於處理各種數值數據，尤其是數組。Python 原生的 list 對象和數組相比，性能和靈活性都有所不足。因此，Numpy 數組得到了廣泛的應用和支持。添加新元素到 Numpy 數組中是常用的操作，這篇文章將會探討如何使用Numpy數組添加元素。

正文

1、使用np.append追加元素

使用numpy的append方法向數組追加元素是非常簡單的。不需要知道數組在內存中的位置。

import numpy as np

arr = np.array([1, 2, 3, 4, 5])
new_arr = np.append(arr, 6)
print(new_arr)

代碼執行後，列印出新數組：[1 2 3 4 5 6]。

2、使用np.insert插入元素

當需要在指定位置插入元素時，可以使用numpy的insert方法。

import numpy as np

arr = np.array([1, 2, 3, 4, 5])
new_arr = np.insert(arr, 2, 9)
print(new_arr)

該代碼塊將元素9插入到了數組索引2位置，執行後列印出新數組：[1 2 9 3 4 5]。

3、使用np.concatenate合併數組

當需要合併兩個數組時，可以使用numpy的concatenate方法。

import numpy as np

arr1 = np.array([1, 2, 3])
arr2 = np.array([4, 5, 6])
new_arr = np.concatenate((arr1, arr2))
print(new_arr)

執行後，列印新數組：[1 2 3 4 5 6]。

4、使用np.vstack和np.hstack堆疊數組

當需要水平或垂直堆疊兩個或多個數組時，可以使用 numpy 的 vstack 和 hstack方法。

水平堆疊

import numpy as np

arr1 = np.array([1, 2, 3])
arr2 = np.array([4, 5, 6])

new_arr = np.hstack((arr1, arr2))
print(new_arr)

執行後，列印新數組：[1 2 3 4 5 6]。

垂直堆疊

import numpy as np

arr1 = np.array([1, 2, 3])
arr2 = np.array([4, 5, 6])

new_arr = np.vstack((arr1, arr2))
print(new_arr)

列印新數組：[[1 2 3] [4 5 6]]。

小結

本文中，我們介紹了向Numpy數組添加元素的若干方法，包括追加元素、插入元素、合併數組以及水平垂直堆疊數組。這些方法對於處理大規模數據非常有用，可以提高代碼效率和可讀性。在使用這些方法時，需要注意數組的維度和數據類型是否匹配等因素。