我正在用 Python 编写一些代码(到目前为止只是为了好玩),这些代码将在 3d 空间中的每个点上存储一些数据。我基本上是在一个存储任意对象的 3d 矩阵对象之后,它允许我做一些高级选择,比如:
- 得到 x=1,y=2,z=3 的点。
- 获取 y=2 的所有点。
- 获取位置 x=1,y=2,z=3 的 3 个单位内的所有点。
- 获取所有points where point.getType() == "Foo"
在上述所有内容中,我需要以某种输出结束,这些输出会给我空间中的原始位置,以及在该点存储的数据。
显然 numpy 可以做我想做的事,但它似乎针对科学计算进行了高度优化,并且到目前为止我还没有弄清楚如何获取我想要的数据。
有没有更好的选择,或者我应该回到用头撞 numpy 的墙? :)
编辑:更多信息前三个答案让我意识到我应该包括:我不担心性能,这纯粹是一个概念验证,我更喜欢干净的代码而不是良好的性能。我还将获得给定 3d 空间中每个点的数据,所以我猜备用矩阵不好吗?
最佳答案
这是另一种常见的方法
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 ]
获取所有points where point.getType() == "Foo"
[ p for p in database if type(p.data) == Foo ]
关于python - 存储 3d 空间中点的信息,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/910930/