Computer >> บทช่วยสอนคอมพิวเตอร์ >  >> การเขียนโปรแกรม >> Python

วิธีเพิ่ม Vector ให้กับ Numpy Array ที่กำหนด


ในปัญหานี้ เราต้องเพิ่ม Vector/Array ให้กับ Numpy Array เราจะกำหนด Numpy Array รวมถึง Vector และบวกพวกมันเข้าด้วยกันเพื่อได้ Array ผลลัพธ์

อัลกอริทึม

ขั้นตอนที่ 1: กำหนด Numpy Array
ขั้นตอนที่ 2: กำหนด Vector
ขั้นตอนที่ 3: สร้าง Result Array เหมือนกับ Array ต้นฉบับ
ขั้นตอนที่ 4: บวก Vector เข้าไปในแต่ละแถวของ Array ต้นฉบับ
ขั้นตอนที่ 5: แสดง 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) สร้าง Array ว่าง 'result' ที่มีมิติเท่ากับ original_array