Trả về bình phương đầu vào giá trị phức trong Python

numpy.square() cũng hoạt động với số phức (complex), trả về bình phương element-wise x*x giữ nguyên shape và dtype complex.

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

Nhập NumPy, tạo mảng số phức 2x2:

import numpy as np

arr = np.array([[3 + 4j, 5 + 7j], [2 + 6j, -2j]])
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)

# 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([[3 + 4j, 5 + 7j], [2 + 6j, -2j]])
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...
 [[ 3.+4.j  5.+7.j]
 [ 2.+6.j -0.-2.j]]
Số chiều...
 2
Kiểu dữ liệu...
 complex128
Kết quả...
 [[ -7.+24.j -24.+70.j]
 [-32.+24.j  -4.+0.j]]

Giải thích: (3+4i)² = 9 + 24i - 16 = -7 + 24i; (5+7i)² = 25 + 70i - 49 = -24 + 70i; v.v.