如何在Quadtree[Python]中读取和使用CSV文件中的坐标？

Question

我正在使用python建立一个Quadtree，并设法为随机生成的点创建一个解决方案，主类是QTree类。

class QTree():
def __init__(self, treshold, customerCount):
    self.threshold = treshold
    self.points = [Point(random.uniform(0, 100), random.uniform(0, 100)) for x in range(customerCount)]
    self.root = Node(0, 0, 100, 100, self.points)

def add_point(x, y):
    self.points.append(Point(x, y))

def get_points(self):
    return self.points

def subdivide(self):
    recursive_subdivide(self.root, self.threshold)

def graph(self):
    fig = plt.figure(figsize=(12, 8))
    x = [point.x for point in self.points]
    y = [point.y for point in self.points]
    ax = fig.add_subplot(111)
    c = find_children(self.root)

    print("\n\nNumber of segments: %d" % len(c))
    areas = set()

    for el in c:
        areas.add(el.width * el.height)
    print("Minimum segment area: %.3f units" % min(areas))

    for n in c:
        ax.add_patch(patches.Rectangle((n.x0, n.y0), n.width, n.height, fill=False))

    plt.title("Quadtree")
    plt.plot(x, y, 'ro', markersize=3, color='b')
    plt.savefig('QuadtreeDiagram.png', dpi=1000)
    plt.show()
    return

我正在获取我的Quadtree图(https:/i.stack.imgur.comojYHO.png。)通过调用Qtree类和其他各种def funcitons。

def test(treshold, customerCount):
    qt = QTree(treshold, customerCount)
    qt.subdivide()
    qt.graph()


 # Tests
   test(1, 50)

我的问题是：我如何改变随机点并使用我自己的CSV文件与坐标？谢谢你 :D

Answer 1

你可以添加一个新的函数来从文件中加载数据并设置点。

 def load_points():

    Df1 = pd.read_csv("D:\\test.txt", sep='\t' )
    self.points=[Point(Df1["Col0"],Df1["Col1"]) for x in range(len(Df1))]

注1: 你必须得到新数据的len和customerCount之间的关系 注2: 我假设文件是这样的:

Col0 Col1

1 2

2 4

3 5