列表解析式

最新推荐文章于 2024-06-21 12:33:31 发布

weixin_43489941

最新推荐文章于 2024-06-21 12:33:31 发布

阅读量396

点赞数 6

文章标签： python

本文链接：https://blog.csdn.net/weixin_43489941/article/details/134814055

版权

import math
def _dist(a, b):
    # Compute Euclidean distance between a and b
    eudis = math.sqrt(math.pow(a[0] - b[0], 2) + math.pow(a[1] - b[1], 2))

    return eudis

indx_list = [0, 1, 2, 3,4]
data_list = []
data_list.append((1,1))
data_list.append((2,2))
data_list.append((3,3))
data_list.append((4,4))
data_list.append((5,5))
T = set()
# for j in indx_list:
#     if len([i for i in indx_list if _dist(data_list[j], data_list[i]) <= 5]) >= 5:
#         T.add(j)
# print("T===",T)

#以下代码等价于注释掉的代码
for j in indx_list:
    list0 = []
    for i in indx_list:
        if _dist(data_list[j], data_list[i]) <= 5:
            list0.append(i)
            if len(list0)>=5:
                T.add(j)

print("T===",T)