Tính căn bậc hai của đầu vào với emath trong Python

Để tính căn bậc hai của các giá trị đầu vào (chỉ chứa số dương), chúng ta sử dụng np.emath.sqrt() (hoặc np.scimath.sqrt()) trong NumPy. Phương thức này hoạt động giống np.sqrt() cho số dương nhưng mở rộng khả năng xử lý số âm và phức.

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

1. Nhập thư viện

import numpy as np

2. Tạo mảng số dương

arr = np.array([1, 4, 9, 16, 25, 36])

3. Hiển thị mảng

print("Mảng của chúng ta...\n", arr)

4. Kiểm tra số chiều

print("\nSố chiều của mảng...\n", arr.ndim)

5. Lấy kiểu dữ liệu

print("\nKiểu dữ liệu của đối tượng mảng...\n", arr.dtype)

6. Lấy hình dạng

print("\nHình dạng của đối tượng mảng...\n", arr.shape)

7. Tính căn bậc hai

print("\nKết quả...\n", np.emath.sqrt(arr))

Ví dụ hoàn chỉnh

import numpy as np

# Tạo mảng NumPy
arr = np.array([1, 4, 9, 16, 25, 36])

# Hiển thị mảng
print("Mảng của chúng ta...\n", arr)

# Kiểm tra số chiều
print("\nSố chiều của mảng...\n", arr.ndim)

# Lấy kiểu dữ liệu
print("\nKiểu dữ liệu của đối tượng mảng...\n", arr.dtype)

# Lấy hình dạng
print("\nHình dạng của đối tượng mảng...\n", arr.shape)

# Tính căn bậc hai với scimath.sqrt()
print("\nKết quả...\n", np.emath.sqrt(arr))

Kết quả đầu ra

Mảng của chúng ta...
 [ 1  4  9 16 25 36]
Số chiều của mảng...
 1
Kiểu dữ liệu của đối tượng mảng...
 int64
Hình dạng của đối tượng mảng...
 (6,)
Kết quả...
 [1. 2. 3. 4. 5. 6.]

Lưu ý: Với mảng chỉ chứa số dương, np.emath.sqrt() trả về kết quả giống hệt np.sqrt() - mảng số thực (float64). Sự khác biệt chỉ hiện rõ khi mảng chứa số âm hoặc phức.