使用坐标计算numpy角度

时间:2019-11-03 17:08:46

标签: python numpy

我必须计算两个点之间的角度,例如A(x1,y1)和B(x2,y2)。我正在使用的当前代码如下-

import math
direction = math.degrees(math.atan((y2 - y1) / (x2 - x1)))

我尝试通过使用以下numpy代码执行相同的代码-

x = np.asarray(data['x'])
y = np.asarray(data['y'])

direction = np.rad2deg(np.arctan2(y, x))

在这里,“ x”和“ y”是指两个具有坐标的属性。

但是,我使用numpy进行方向计算所获得的值与使用'math'软件包所获得的计算结果有所不同。

只需提供最小值和最大值-

data['x'].min(), data['x'].max()                                       
# (25.24, 803.85)

data['y'].min(), data['y'].max()                                       
# (21.44, 805.76)

“ x”和“ y”的两个分布几乎具有正态分布。

如何使用numpy进行角度计算?

谢谢!

样本数据-

data_d = 
{'time': [3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13],
 'x': [405.31,
  405.3,
  405.29,
  405.27,
  405.27,
  405.27,
  405.31,
  405.38,
  405.46,
  405.54,
  405.63],
 'y': [417.07,
  416.86,
  416.71,
  416.61,
  416.54,
  416.49,
  416.37,
  416.27,
  416.13,
  415.93,
  415.84],
 'direction': [87.27368900609596,
  86.18592516571398,
  78.69006752595475,
  0.0,
  0.0,
  -71.56505117706985,
  -55.007979801450084,
  -60.25511870306028,
  -68.19859051363696,
  -45.00000000001809,
  -67.38013505194608],
 'direction_np_computation': [-134.1807285626706,
  -134.19444442862144,
  -134.2040441491018,
  -134.2095039258752,
  -134.21431600901414,
  -134.2177537199761,
  -134.2288323248442,
  -134.24065659640502,
  -134.2559407971113,
  -134.27535831135458,
  -134.2879109537407]}

data = pd.DataFrame(data_d)

在这里,使用“数学”包计算“方向”列,并使用代码计算“ direction_np_computation”-

# Reference point- Xr = 0 Yr = 0


# Get 'x' and 'y' attributes from 'data'- x = np.array(data['x']) y = np.array(data['y'])

# Compute direction from reference point with the coordinates- direction = np.rad2deg(np.arctan2(Yr - y, Xr - x))

为什么两列之间不匹配?

1 个答案:

答案 0 :(得分:1)

在我看来,它运作良好。由于我看不到您的字符numpy / pandas坐标数组,因此我无法为您提供确切的解决方案

3个版本:

arctan

>>> direction = np.rad2deg(np.arctan((2-1)/(2-1)))
>>> direction
45.0

数学

>>> direction = math.degrees(math.atan((2 - 1) / (2 - 1)))
>>> direction
45.0

arctan2

>>> direction = np.rad2deg(np.arctan2((2-1),(2-1)))
>>> direction
45.0

使用虚假数据(我看不到您的数据),使用您的功能

参考点:

>>> Xr = 10
>>> Yr = 10

使用以下坐标检查指向参考点的方向:

>>> Xs = np.array([1,2,3,4,5,6])
>>> Ys = np.array([1,2,3,4,5,6])

我希望它们的方向都为45度

>>> direction = np.rad2deg(np.arctan2(Yr-Ys,Xr-Xs))
>>> direction
array([45., 45., 45., 45., 45., 45.])

编辑

根据您提供的数据:

您似乎对结果不同的想法是正确的。

此处有更多信息:numpy arctan2 bug or usage issues?

使用np.arctan应该没问题

numpy模块

>>> direction = np.rad2deg(np.arctan(data.y/data.x))
>>> direction
0     45.819271
1     45.805556
2     45.795956
3     45.790496
4     45.785684
5     45.782246
6     45.771168
7     45.759343
8     45.744059
9     45.724642
10    45.712089
dtype: float64

数学模块

>>> for i in range(10):
...     math.degrees(math.atan((data.y[i]) / (data.x[i])))
... 
45.81927143732942
45.805555571378584
45.79595585089821
45.79049607412479
45.78568399098587
45.782246280023934
45.771167675155795
45.75934340359498
45.744059202888714
45.72464168864543
>>> 

注意

请注意符号+-。 Y2-Y1 / X2-X1与Y1-Y2 / X1-X2可以给出不同的结果,但实际上两者都是正确的。在我们的情况下,这些结果为45度或-135度。他们俩都是正确的,只是其中之一是顺时针的