如何使用Swift中的PythonKit将布尔掩码数组与NumPy一起使用

Question

我有以下Python代码：

print("Cluster connected triangles")
with o3d.utility.VerbosityContextManager(
o3d.utility.VerbosityLevel.Debug) as cm:
triangle_clusters, cluster_n_triangles, cluster_area = (
mesh.cluster_connected_triangles())
triangle_clusters = np.asarray(triangle_clusters)
cluster_n_triangles = np.asarray(cluster_n_triangles)
cluster_area = np.asarray(cluster_area)
print("Get mesh with small clusters removed")
mesh_1 = copy.deepcopy(mesh)
largest_cluster_idx = cluster_n_triangles.argmax()
triangles_to_remove = triangle_clusters != largest_cluster_idx
mesh_1.remove_triangles_by_mask(triangles_to_remove)

我正试图在Swift(PythonKit(中进行改编

let np = Python.import("numpy")
let clusters = mesh.cluster_connected_triangles()
let triangle_clusters = np.asarray(clusters[0])
let cluster_n_triangles = np.asarray(clusters[1])
let cluster_area = np.asarray(clusters[2])
let largest_cluster_idx = cluster_n_triangles.argmax()
let triangles_to_remove = (triangle_clusters != largest_cluster_idx).  // ERROR: Ambiguous use of operator '!='
let largestCluster = mesh.remove_triangles_by_mask(triangles_to_remove)

有人能帮我转换这行Python吗：Swift/PythonKit 中的triangles_to_remove = triangle_clusters != largest_cluster_idx

换句话说：创建布尔掩码数组的PythonKit/Swift语法是什么？

Answer 1

答案是将代码包装在PythonObject:中

print("Clustering...")
let clusters = mesh.cluster_connected_triangles()
let triangle_clusters = np.asarray(clusters[0])
let cluster_n_triangles = np.asarray(clusters[1])
let cluster_area = np.asarray(clusters[2])
let largest_cluster_idx = cluster_n_triangles.argmax()
print("(mesh) : with (cluster_n_triangles.count) clusters.")
print("Largest cluster : (cluster_n_triangles[largest_cluster_idx]) triangles.")
//THIS!
let triangles_to_remove = PythonObject(triangle_clusters != largest_cluster_idx)
mesh.remove_triangles_by_mask(triangles_to_remove)
mesh.remove_unreferenced_vertices()
print("Cluster Result: (mesh)")