Computer >> Máy Tính >  >> Lập trình >> Python

Làm thế nào để thêm một vectơ vào một mảng Numpy đã cho?

Trong bài toán này, chúng ta phải thêm một vectơ / mảng vào một mảng numpy. Chúng tôi sẽ xác định mảng numpy cũng như vectơ và thêm chúng để có được mảng kết quả

Thuật toán

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.

Mã mẫu

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)

Đầu ra

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]]

Giải thích

Câu lệnh result =np.empty_like (original_array) tạo ra một 'kết quả' mảng trống có cùng kích thước với original_array