同一列表的多次迭代

时间:2015-06-14 21:00:19

标签: python python-2.7 python-3.x numpy shapely

我有一个数据列表如下:

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

1 个答案:

答案 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元素。第二个子列表包含错误的元素。

您的问题有点令人困惑,datasublists似乎包含相同的元素。所以,也许你没有测试A是否在data中(或与data的元素相交),但是A是否与[A,B,C]的另一个元素相交{1}}等等。

在任何情况下,收集结果的关键是在results中有两个(或更多)插槽,您可以根据测试放置iresults也可以是字典,也可以是两个不同的变量。例如results={'found':[],'lost':[]}

我们是否需要在测试中更多地工作?

A 'intersects with' any of [B,C]
B 'intersects with' any of [A,C]
C 'intersects with' any of [A,B]