Computer >> Hướng Dẫn Máy Tính >  >> Lập Trình >> Python

Cách thêm vector vào mảng NumPy cho trước


Trong bài này, cộng một vector/mảng vào mảng numpy. Định nghĩa mảng numpy và vector, cộng chúng để có mảng kết quả.

Thuật toán

Bước 1: Định nghĩa mảng numpy.
Bước 2: Định nghĩa vector.
Bước 3: Tạo mảng kết quả cùng shape mảng gốc.
Bước 4: Cộng vector vào từng hàng của mảng gốc.
Bước 5: In mảng kết quả.

Mã ví dụ

import numpy as np
original_array = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9], [10, 11, 12]])
print("Mảng gốc: \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("\nKết quả: \n", result)

Kết quả

Mảng gốc:  [[ 1  2  3]  [ 4  5  6]  [ 7  8  9]  [10 11 12]]
Vector:  [1 1 0]
Kết quả:  [[ 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 mảng rỗng 'result' cùng kích thước mảng gốc.