如何用Python将Pandas点云数据直接存储为glTF文件

依赖安装 #

先安装所需的Python库：

pip install pygltflib pandas numpy

实现代码 #

以下是完整的实现流程，可直接替换为你自己的DataFrame数据：

import pandas as pd
import numpy as np
from pygltflib import GLTF2, Buffer, BufferView, Accessor, Mesh, Primitive, Node, Scene, Asset

# 1. 替换为你自己的DataFrame数据
data = {
    'x': [1.0, 2.0, 3.0, 4.0],
    'y': [0.0, 1.0, 2.0, 3.0],
    'z': [0.0, 0.0, 0.0, 0.0],
    'r': [255, 0, 0, 255],
    'g': [0, 255, 0, 255],
    'b': [0, 0, 255, 0]
}
df = pd.DataFrame(data)

# 2. 转换数据为glTF兼容的二进制格式
# 坐标转为float32字节数组
positions = df[['x', 'y', 'z']].to_numpy(dtype=np.float32).tobytes()
# 颜色转为uint8字节数组（若你的r/g/b是0-1浮点数，需先执行：(df[['r','g','b']]*255).astype(np.uint8)）
colors = df[['r', 'g', 'b']].to_numpy(dtype=np.uint8).tobytes()

# 3. 构建glTF核心结构
gltf = GLTF2()
# 设置资产元信息
gltf.asset = Asset(version="2.0", generator="Custom Point Cloud Exporter")

# 创建缓冲区存储二进制数据
buffer = Buffer(uri=None, byteLength=len(positions) + len(colors))
gltf.buffers.append(buffer)

# 创建缓冲视图，分别映射坐标和颜色数据
pos_view = BufferView(
    buffer=0,
    byteOffset=0,
    byteLength=len(positions),
    target=34962  # ARRAY_BUFFER，指定用于顶点数据
)
color_view = BufferView(
    buffer=0,
    byteOffset=len(positions),
    byteLength=len(colors),
    target=34962
)
gltf.bufferViews.extend([pos_view, color_view])

# 创建访问器，描述数据的类型和结构
pos_accessor = Accessor(
    bufferView=0,
    byteOffset=0,
    componentType=5126,  # FLOAT类型
    count=len(df),
    type="VEC3",
    min=df[['x', 'y', 'z']].min().tolist(),
    max=df[['x', 'y', 'z']].max().tolist()
)
color_accessor = Accessor(
    bufferView=1,
    byteOffset=0,
    componentType=5121,  # UNSIGNED_BYTE类型
    count=len(df),
    type="VEC3",
    normalized=True  # 标记为归一化，glTF会自动将0-255转为0-1的颜色值
)
gltf.accessors.extend([pos_accessor, color_accessor])

# 创建点云网格：使用POINTS模式
primitive = Primitive(
    attributes={"POSITION": 0, "COLOR_0": 1},
    mode=0  # 0对应glTF规范中的POINTS渲染模式
)
mesh = Mesh(primitives=[primitive])
gltf.meshes.append(mesh)

# 创建节点和场景
node = Node(mesh=0)
gltf.nodes.append(node)
scene = Scene(nodes=[0])
gltf.scenes.append(scene)
gltf.scene = 0  # 设置默认场景

# 4. 保存glTF文件（内嵌二进制数据，生成单个文件）
gltf.set_binary_blob(positions + colors)
gltf.save("point_cloud.gltf")

关键细节说明 #

• POINTS模式：mode=0是glTF 2.0指定的点云渲染模式，无需将点转换为三角网格。
• 颜色数据适配：如果你的DataFrame中r/g/b是0-1范围的浮点数，需要先乘以255转换为0-255的整数，再转为uint8类型。
• 数据存储方式：示例将二进制数据内嵌到.gltf文件中，若需要外部.bin文件，可将buffer.uri设为"point_cloud.bin"，并单独保存positions + colors到该文件。

内容的提问来源于stack exchange，提问作者Daniel van der Maas