如何使用cloudComPy为点云的自定义标量字段赋值？

问题描述 #

我有一组点数组需要转换为点云，正在使用CloudCompare的Python API——cloudComPy实现该需求。最终目标是用这些点创建点云对象，并为其添加名为“lane_id”、值为2的标量字段。

我已经通过以下代码创建了点云：

import cloudComPy as cc
import numpy as np

# 示例点数组
arr_of_points = np.arange(1,31).reshape(-1, 3)
cloud = cc.ccPointCloud()
cloud.coordsFromNPArray_copy(arr_of_points)

成功创建点云后，我用cloud.addScalarField("lane_id")添加了标量字段，但不知道如何给这个字段赋值。

解决方案 #

要给新建的标量字段赋值，需要先获取该标量字段对象，再通过numpy数组批量赋值，具体步骤如下：

1. 获取标量字段对象
   可以通过字段名称或索引获取刚添加的标量字段：

   # 通过名称获取
   sf = cloud.getScalarField("lane_id")
   # 或通过索引获取（刚添加的字段是最后一个，索引为总数量-1）
   # sf = cloud.getScalarField(cloud.getNumberOfScalarFields() - 1)
   

2. 生成赋值数组
   创建一个和点云点数一致的numpy数组，所有元素设为2：

   num_points = cloud.size()
   lane_id_values = np.full(num_points, 2, dtype=np.float32)  # cloudComPy标量字段默认使用float32类型
   

3. 写入标量字段
   调用标量字段的fromNPArray_copy方法完成赋值：

   sf.fromNPArray_copy(lane_id_values)
   

完整代码 #

import cloudComPy as cc
import numpy as np

# 示例点数组
arr_of_points = np.arange(1,31).reshape(-1, 3)
cloud = cc.ccPointCloud()
cloud.coordsFromNPArray_copy(arr_of_points)

# 添加标量字段并赋值
cloud.addScalarField("lane_id")
sf = cloud.getScalarField("lane_id")
num_points = cloud.size()
lane_id_values = np.full(num_points, 2, dtype=np.float32)
sf.fromNPArray_copy(lane_id_values)

# 可选：将该标量字段设为当前活跃字段
cloud.setCurrentScalarField(cloud.getScalarFieldIndexByName("lane_id"))

内容的提问来源于stack exchange，提问作者rr_goyal