Python numpy arctan2用法及代碼示例

本文簡要介紹 python 語言中 numpy.arctan2 的用法。

用法:

numpy.arctan2(x1, x2, /, out=None, *, where=True, casting='same_kind', order='K', dtype=None, subok=True [, signature, extobj ]) = <ufunc 'arctan2'>

x1/x2 的元素級反正切正確選擇象限。

象限(即分支)的選擇使得arctan2(x1, x2)是在原點結束並通過點 (1,0) 的射線與在原點結束並通過點 (x2,x1)。 (注意角色互換：“y-coordinate”是第一個函數參數，“x-coordinate” 是第二個。)按照 IEEE 約定，此函數定義為x2= +/-0 並且對於其中一個或兩個x1和x2= +/-inf(具體值見注釋)。

該函數沒有為 complex-valued 參數定義；對於所謂的複數值參數，請使用 angle 。

參數：

x1： 類數組，實值

y 坐標。

x2： 類數組，實值

x-坐標。如果x1.shape != x2.shape，它們必須可以廣播到一個共同的形狀(成為輸出的形狀)。

out： ndarray，None，或 ndarray 和 None 的元組，可選

存儲結果的位置。如果提供，它必須具有輸入廣播到的形狀。如果未提供或 None，則返回一個新分配的數組。元組(隻能作為關鍵字參數)的長度必須等於輸出的數量。

where： 數組，可選

此條件通過輸入廣播。在條件為真的位置，out數組將設置為 ufunc 結果。在其他地方，out數組將保留其原始值。請注意，如果未初始化out數組是通過默認創建的out=None，其中條件為 False 的位置將保持未初始化狀態。

**kwargs：

對於其他僅關鍵字參數，請參閱 ufunc 文檔。

返回：

angle： ndarray

以弧度為單位的角度數組，在範圍內[-pi, pi].如果兩者都是標量x1和x2是標量。

注意：

arctan2與atan2底層 C 庫的函數。 C 標準中定義了以下特殊值：[1]

x1	x2	arctan2(x1,x2)
+/- 0	+0	+/- 0
+/- 0	-0	+/- 圓周率
> 0	+/-inf	+0 /+pi
< 0	+/-inf	-0 /-pi
+/-inf	+信息	+/- (pi/4)
+/-inf	-inf	+/- (3*pi/4)

請注意，+0 和 -0 是不同的浮點數，+inf 和 -inf 也是如此。

參考：

1：

ISO/IEC 標準 9899:1999，“編程語言 C”。

例子：

考慮不同象限中的四個點：

>>> x = np.array([-1, +1, +1, -1])
>>> y = np.array([-1, -1, +1, +1])
>>> np.arctan2(y, x) * 180 / np.pi
array([-135.,  -45.,   45.,  135.])

注意參數的順序。arctan2也被定義為x2= 0 和其他幾個特殊點，獲取範圍內的值[-pi, pi]：

>>> np.arctan2([1., -1.], [0., 0.])
array([ 1.57079633, -1.57079633])
>>> np.arctan2([0., 0., np.inf], [+0., -0., np.inf])
array([ 0.        ,  3.14159265,  0.78539816])