在 3d 空间中存储点的信息

Question

我正在用 Python 编写一些代码（到目前为止只是为了好玩），它将在 3d 空间中的每个点上存储一些数据。我基本上是在存储任意对象的 3d 矩阵对象之后，这些对象将允许我进行一些高级选择，例如：

• 得到 x=1,y=2,z=3 的点。
• 获取 y=2 的所有点。
• 获取位置 x=1,y=2,z=3 的 3 个单位内的所有点。
• 获取 point.getType() == "Foo" 的所有点

在上述所有内容中，我需要最终得到某种输出，该输出将为我提供空间中的原始位置以及存储在该点的数据。

显然 numpy 可以做我想做的事，但它似乎对科学计算进行了高度优化，并且到目前为止我还没有弄清楚如何获得我想要的数据。

有没有更好的选择，还是我应该回到麻木的墙上敲我的头？:)

编辑：前三个答案让我意识到我应该包括更多信息：我不担心性能，这纯粹是一个概念验证，我更喜欢干净的代码而不是良好的性能。我还将拥有给定 3d 空间中每个点的数据，所以我猜备用矩阵是坏的？

Answer 1

这是另一种常见的方法

class Point( object ):
    def __init__( self, x, y, z, data ):
        self.x, self.y, self.z = x, y, z
        self.data = data
    def distFrom( self, x, y, z )
        return math.sqrt( (self.x-x)**2 + (self.y-y)**2 + (self.z-z)**2 )

database = [ Point(x,y,z,data), Point(x,y,z,data), ... ]

让我们看看您的用例。

得到 x=1,y=2,z=3 的点。

[ p for p in database if (p.x, p.y, p.z) == ( 1, 2, 3 ) ]

获取 y=2 的所有点。

[ p for p in database if p.y == 2 ]

获取位置 x=1,y=2,z=3 的 3 个单位内的所有点。

[ p for p in database if p.distFrom( 1, 2, 3 ) <= 3.0 ]

获取 point.getType() == "Foo" 的所有点

[ p for p in database if type(p.data) == Foo ]