python - 将世界分成垃圾箱

时间:2016-03-10 01:30:24

标签: python algorithm data-structures bins

我正在努力将移动球放入适当的垃圾箱。我喜欢认为我走在正确的轨道上,但我现在已经被卡住了一段时间。

我留下了与我的问题无关的代码,但如果回答的人需要进一步的细节,我可以提供。基本上,我有一个200个移动球的世界。它们具有X和Y坐标。我想将世界划分为宽度为256的方形箱,并将球放在适当的箱子中。

我的方法是将它们放入字典中。它看起来像这样:

dict_of_balls = {}
for i in range(len(balls)):
    xb = int(balls[i].x/256)
    yb = int(balls[i].y/256)

我想让键成为(xb, yb)对的元组,然后将适当的球放在那个箱子中,但我不认为你可以使用元组作为键......

代码如下:

import math
import random
import time
import sys


ball_min_radius = 16.0 #world coordinates         
ball_max_radius = 128.0  #world coordniates
number_balls = 200

class Ball:
    """ 
    Implements a point/ball
    """

    def __init__(self):
          self.x = random.uniform(world_min_x,world_max_x)
          self.y = random.uniform(world_min_y,world_max_y)
          self.radius = int(random.uniform(ball_min_radius,ball_max_radius))
    def __lt__(self, other):
        return self.id < other.id

def main():
    world_min_x = -200.0*number_balls**.5  # minimum x in world coordinates
    world_max_x = +200.0*number_balls**.5  # maximum x in world coordinates
    world_min_y = -200.0*number_balls**.5  # minimum y in world coordinates
    world_max_y = +200.0*number_balls**.5  # maximum y in world coordinates

    balls = [Ball() for i in range(number_balls)]

对于如何根据给定的世界坐标将世界分成垃圾箱,有没有人有任何想法?我不确定要使用哪种数据结构,因为我不能使用元组作为密钥。提前感谢您的任何反馈。

2 个答案:

答案 0 :(得分:1)

你为什么想要一本字典?以下是你将如何做到这一点,但请记住,你只会 每个bin获得一个球,因为你专门将它们的键转换为(int,int)并且键是唯一的。

如果您使用集合,您还可以排序(在我的示例中,我按区域标识符排序):

我不确定你在做什么,但你可以做到:

import math
import random
import time
import sys


ball_min_radius = 16.0 #world coordinates         
ball_max_radius = 128.0  #world coordniates
number_balls = 200

world_min_x = -200.0*number_balls**.5  # minimum x in world coordinates
world_max_x = +200.0*number_balls**.5  # maximum x in world coordinates
world_min_y = -200.0*number_balls**.5  # minimum y in world coordinates
world_max_y = +200.0*number_balls**.5  # maximum y in world coordinates


class Ball:
    """ 
    Implements a point/ball
    """

    def __init__(self):
          self.x = random.uniform(world_min_x,world_max_x)
          self.y = random.uniform(world_min_y,world_max_y)
          self.radius = int(random.uniform(ball_min_radius,ball_max_radius))
    def __lt__(self, other):
        return self.id < other.id

    def __str__(self):
        return 'x={x} y={y} r={r}'.format(x=self.x, y=self.y, r=self.radius)

def main():

    balls = [Ball() for i in range(number_balls)]

    dict_of_balls = {}
    ball_collection = []
    for b in balls:
        xb = int(b.x/256)
        yb = int(b.y/256)
        key = (xb, yb)
        dict_of_balls[key] = b

        ball_collection.append((key, b))

    print 'length of dictionary:{}'.format(len(dict_of_balls.keys()))
    print 'length of collection:{}'.format(len(ball_collection))

请注意,字典的项目数少于集合。

您还可以通过这种方式轻松打印每件商品:

    for b in ball_collection:
        print 'ball region: {r}   with coords: {c}'.format(r=b[0], c=b[1])

或者,如果您愿意,可以对它们进行排序:

    print 'Collections also let you sort the collection by region(s)...'
    sorted_list = sorted(ball_collection, key= lambda x: (x[0][0], x[0][1]))


    for b in sorted_list:
        print 'ball region: {r}   with coords: {c}'.format(r=b[0], c=b[1])

你也可以在特定地区轻松获得球:

    print '... or get only ones in a specific region'
    subset =  [b for b in ball_collection if b[0][0] == 1]

    for b in subset:
        print 'ball region: {r}   with coords: {c}'.format(r=b[0], c=b[1])


main()

一个集合似乎可以实现实际想要的东西。

答案 1 :(得分:0)

您可以在字典中使用元组作为键,因为元组是不可变的。您不能用于字典键的唯一数据类型是列表[]或设置{}

**a = {(1,2):'example1', (2,3):'example2'}
>>> a[(1,2)]
'example1'**

所以我相信这会让你更容易解决问题。