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

Tính arc tangent element-wise x1/x2 chọn quadrant đúng trong Python

numpy.arctan2(y, x) tính góc signed angle (radian) giữa tia x dương và điểm (x, y), chọn quadrant đúng dựa trên dấu của x, y. Kết quả trong [-π, π]. Khác với arctan(y/x) chỉ trả về [-π/2, π/2].

Tham số

  • y (array_like): Hoành độ (y-coordinates).
  • x (array_like): Tung độ (x-coordinates). Phải broadcast được cùng shape.

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

Nhập NumPy, tạo 4 điểm 4 quadrant:

import numpy as np

x = np.array([-1, +1, +1, -1])  # x-coordinates
y = np.array([-1, -1, +1, +1])  # y-coordinates

print("Tọa độ x...\n", x)
print("\nTọa độ y...\n", y)

# Tính góc độ (radian -> degree)
print("\nKết quả (độ)...\n", np.arctan2(y, x) * 180 / np.pi)

Ví dụ hoàn chỉnh

import numpy as np

x = np.array([-1, +1, +1, -1])
y = np.array([-1, -1, +1, +1])
print("Tọa độ x...\n", x)
print("\nTọa độ y...\n", y)
print("\nKết quả (độ)...\n", np.arctan2(y, x) * 180 / np.pi)

Kết quả đầu ra

Tọa độ x...
 [-1  1  1 -1]
Tọa độ y...
 [-1 -1  1  1]
Kết quả (độ)...
 [-135.  -45.   45.  135.]

Giải thích: (-1,-1) Quadrant III -> -135°; (1,-1) Quadrant IV -> -45°; (1,1) Quadrant I -> 45°; (-1,1) Quadrant II -> 135°.