计算 Python 中两个大列表的每个元素的距离

Question

我有两个大的元组列表（大小 > 5000），其中包含如下坐标：

l1 = [ (33.5779, -111.925),
 (33.5738, -111.906),
 (33.5556, -111.899),
...
 (33.5646, -111.915),
 (33.5733, -111.863)]

l2 = [ (33.4318, -111.938),
 (33.5228, -111.9),
 (33.5387, -111.885),
...
 (33.5264, -111.925),
 (33.538, -111.888)]

我想找出符合以下条件的组合：

distance <= arbitraryDistance

距离是：

 from geopy.distance import great_circle
 great_circle((longitude1,latitude1), (longitude2,latitude2))

最快的方法是什么？

Answer 1

我找到的一个相对较快的解决方案是使用来自 scipy.spatial.distance 的函数cdist

它计算两个列表中每个点组合之间的欧几里得距离。即使它不计算以米为单位的距离，但以度为单位，我仍然可以将我的米要求转换为度数。

所以我目前的解决方案是这样的：

from scipy.spatial.distance import cdist
l1 = [ (33.5779, -111.925),
 (33.5738, -111.906),
 (33.5556, -111.899),
...
 (33.5646, -111.915),
 (33.5733, -111.863)]

l2 = [ (33.4318, -111.938),
 (33.5228, -111.9),
 (33.5387, -111.885),
...
 (33.5264, -111.925),
 (33.538, -111.888)]

distanceRequired = 0.02 #arbitrary number in degrees

matrixOfDistances = cdist(l1,l2)
for index1,value1 in enumerate(matrixOfDistances):
  isABusinessOpportunity = True
  for index2,value2 in enumerate(value1):
    if value2 <= distanceRequired:
      isABusinessOpportunity = False
        break

  if isABusinessOpportunity:
    print "The combination is:" + str(index1) + str(index2)

Answer 2

我认为列表理解在这里会很好用

Result = []
Result = [great_circle(x,y) for x in l1 for y in l2 if great_circle(x,y) <= ArbitraryDist]

print(Result)

试试这个

Answer 3

你的回答有两个意思，我都会回答。

您想要列表一中的元素 1 和列表二中的元素 2 的距离：

我会使用Geopandas 来完成此操作，它是Shapely 和pandas 的混搭。 Shapely 很像 geopy，而 pandas 是一个用于进行矢量化计算的数学库。您的问题非常适合矢量化，因此您可以获得一些非常好的加速。

Shapely 中两点之间的距离可以这样计算：

from shapely.geometry import Point

Point(1,1).distance(Point(4,3))
3.605551275463989

从两个列表中创建两个 GeoSeries，其中 x、y 的每个元组都应该是一个 Shapely Point。

然后您可以使用本机 GeoSeries.distance 方法，该方法采用您的另一个 GeoSeries 并返回距离列表。

from geopandas import GeoSeries

s1 = GeoSeries([Point(1,1), Point(2,1)])
s2 = GeoSeries([Point(4,3), Point(2,5)])

s1.distance(s2)
0    3.605551
1    4.000000
dtype: float64

有了这个GeoSeries的距离，你可以简单地说

gs_dist = s1.distance(s2)
mask = gs_dist < 4

您将收到True / False 的列表，其中匹配条件。

您可以使用此蒙版切入您的 GeoSeries，如下所示：

s1[mask]
0    POINT (1 1)
dtype: object

或者你想要两个列表中每个元素和所有元素之间的距离

由于两个包含 5000 个对象的列表的组合往往会在内存中爆炸，因此您可能需要使用生成器。

itertools 有一个很好的组合生成器。

使用itertools.izip 以生成器的方式获取所有元素的所有可能对：

for pair in itertools.izip(l1, l2):
   point1, point2 = pair
   distance = point1.distance(point2)