如何將一個向量新增到給定的 Numpy 陣列?


在這個問題中,我們必須將一個向量/陣列新增到一個 numpy 陣列中。我們將定義 numpy 陣列和向量,並將它們相加得到結果陣列

演算法

Step 1: Define a numpy array.
Step 2: Define a vector.
Step 3: Create a result array same as the original array.
Step 4: Add vector to each row of the original array.
Step 5: Print the result array.

示例程式碼

import numpy as np

original_array = np.array([[1,2,3], [4,5,6], [7,8,9], [10, 11, 12]])
print("Original Array: \n", original_array)

vector = np.array([1,1,0])
print("\nVector: ", vector)
result = np.empty_like(original_array)
for i in range(4):
   result[i,:] = original_array[i,:] + vector
print("\nResult: \n", result)

輸出

Original Array:
 [[ 1  2  3]
 [ 4  5  6]
 [ 7  8  9]
 [10 11 12]]
Vector:  [1 1 0]

Result:
 [[ 2  3  3]
 [ 5  6  6]
 [ 8  9  9]
 [11 12 12]]

解釋

語句 result = np.empty_like(original_array) 建立了一個與 original_array 具有相同維度大小的空陣列 'result'

更新於: 16-Mar-2021

3K+ 瀏覽

開啟您的 職業生涯

透過完成課程獲得認證

開始
廣告
© . All rights reserved.