Trả về bình phương từng phần tử mảng trong Python

Để tính bình phương element-wise (từng phần tử) của mảng, dùng numpy.square(). Trả về x*x cùng hình dạng và dtype với đầu vào. Scalar đầu vào cho ra scalar.

Tham số

• x (array_like): Dữ liệu đầu vào.
• out (ndarray, tùy chọn): Vị trí lưu kết quả.
• where (array_like, tùy chọn): Điều kiện broadcast.

Các bước thực hiện

Nhập NumPy, tạo mảng 2D int:

import numpy as np

arr = np.array([[25, -50, 75], [-90, 81, 64]])
print("Mảng của chúng ta...\n", arr)
print("\nSố chiều...\n", arr.ndim)
print("\nKiểu dữ liệu...\n", arr.dtype)

# Tính bình phương element-wise
print("\nKết quả...\n", np.square(arr))

Ví dụ hoàn chỉnh

import numpy as np

arr = np.array([[25, -50, 75], [-90, 81, 64]])
print("Mảng của chúng ta...\n", arr)
print("\nSố chiều...\n", arr.ndim)
print("\nKiểu dữ liệu...\n", arr.dtype)
print("\nKết quả...\n", np.square(arr))

Kết quả đầu ra

Mảng của chúng ta...
 [[ 25 -50  75]
 [-90  81  64]]
Số chiều...
 2
Kiểu dữ liệu...
 int64
Kết quả...
 [[ 625 2500 5625]
 [8100 6561 4096]]