Python：将二维点云转换为灰度图像

Question

我有一个长度可变的数组，其中填充了分布在 (0,0) 附近的二维坐标点（来自点云），我想将它们转换为二维矩阵（=灰度图像）。

# have
array = [(1.0,1.1),(0.0,0.0),...]
# want
matrix = [[0,100,...],[255,255,...],...]

我将如何使用 python 和 numpy 实现这一点

Answer 1

看起来matplotlib.pyplot.hist2d就是你要找的。

它基本上将您的数据分箱为二维分箱（大小由您选择）。 这里的文档和工作示例如下。

import numpy as np
import matplotlib.pyplot as plt
data = [np.random.randn(1000), np.random.randn(1000)]
plt.scatter(data[0], data[1])

然后你可以在你的数据上调用hist2d ，例如这样

plt.hist2d(data[0], data[1], bins=20)

请注意， hist2d的参数是两个一维数组，因此在将数据提供给hist2d之前，您必须对数据进行一些hist2d 。

Answer 2

仅使用 numpy 的快速解决方案，无需 matplotlib 及其绘图：

import numpy as np
# given a 2dArray "array" and a desired image shape "[x,y]"
matrix = np.histogram2d(array[:,0], array[:,1], bins=[x,y])