我有一个数据列表如下:
from shapely.geometry import box
data = [box(1,2,3,4), box(4,5,6,7), box(1,2,3,4)]
sublists = [A,B,C]
列表'数据'有以下子列表:
A = box(1,2,3,4)
B = box(4,5,6,7)
C = box(1,2,3,4)
我必须检查子列表是否相交。如果相交,他们应该放入一个元组;如果不相交应该放入不同的元组。预期结果是:
result = [(A,C), (B)]
怎么做?
我试过它:
results = []
for p,c in zip(data,sub_lists):
for x in data:
if p.intersects(x): ##.intersects return true if they overlap else false
results.append(c)
print results
答案 0 :(得分:1)
如果不下载shapely
,我认为您要对列表执行的操作可以使用字符串(或整数)进行复制:
In [221]: data=['one','two','three']
In [222]: data1=['one','four','two']
In [223]: results=[[],[]]
In [224]: for i in data1:
if i in data:
results[0].append(i)
else:
results[1].append(i)
.....:
In [225]: results
Out[225]: [['one', 'two'], ['four']]
将i in data
替换为intersects
测试。 results
的第一个子列表包含测试为真的data1
元素。第二个子列表包含错误的元素。
您的问题有点令人困惑,data
和sublists
似乎包含相同的元素。所以,也许你没有测试A
是否在data
中(或与data
的元素相交),但是A
是否与[A,B,C]
的另一个元素相交{1}}等等。
在任何情况下,收集结果的关键是在results
中有两个(或更多)插槽,您可以根据测试放置i
。 results
也可以是字典,也可以是两个不同的变量。例如results={'found':[],'lost':[]}
。
我们是否需要在测试中更多地工作?
A 'intersects with' any of [B,C]
B 'intersects with' any of [A,C]
C 'intersects with' any of [A,B]