Python – 3D 矩阵到坐标列表

如果您需要将一个三维矩阵转化为坐标列表，那么可以使用Python中的numpy库。下面是一个示例程序，说明如何使用numpy库将三维矩阵转化为坐标列表。

示例程序

import numpy as np

# 定义一个三维矩阵
matrix = np.array([
  [[1,2],[3,4]],
  [[5,6],[7,8]],
  [[9,10],[11,12]]
])

# 使用numpy库将矩阵转化为坐标列表
coordinates = np.column_stack(np.where(matrix > 0)).tolist()

# 打印结果
print(coordinates)

输出结果

[[0, 0, 0], [0, 1, 1], [1, 0, 2], [1, 1, 3], [2, 0, 4], [2, 1, 5]]

解释

在上面的程序中，我们首先定义了一个三维矩阵。然后，我们使用numpy库中的where()函数来查找矩阵中大于零的元素的位置。这将返回一个包含所有大于零元素的索引值的元组。最后，我们使用column_stack()函数将这些索引值连接成一个坐标列表，并使用tolist()函数将其转化为Python列表。