python - 计算网格框中的点

Question

假设有一个网格，其中有一些点，如下图所示。 我的目标是计算每格网格的点数。这是我的第一次尝试。

            for tupel in point_list:
               a=0
               b=0
               for i in self.boxvector:
                  if tupel[0] < i:
                     a=self.boxvector.index(i)-1
                     break

               for i in self.boxvector:
                  if tupel[1] < i:
                     b=self.boxvector.index(i)-1
                     break

               farray[a][b]+=1

它有效，但速度很慢。有没有加快一点的？

我使用名为 boxvector 的变量来定义网格。在这个例子中，boxvector 是:boxvector = [-1., -.5, 0, .5, 1.]。网格始终是最大值为 -1 和 1 的二次方。这些框通过 farray 表示，看起来像 farray = [[0,0,0,0],[0,0,0,0],[0,0,0, 0],[0,0,0,0]]。因此，每次算法在相应的框中找到一个点时，每个框都有一个值，该值会递增。 point_list 的形式为 point_list = [(x0,y0),(x1,y1),(x3,y3), ...]

感谢您的帮助!

Answer 1

鉴于您似乎已经在使用 matplotlib，只需使用 numpy.histogram2d .

举个例子:

import numpy as np
import matplotlib.pyplot as plt

t = np.linspace(0, 4*np.pi, 100)
x = np.cos(3 * t)
y = np.sin(t)

gridx = np.linspace(-1, 1, 5)
gridy = np.linspace(-1, 1, 5)

grid, _, _ = np.histogram2d(x, y, bins=[gridx, gridy])

plt.figure()
plt.plot(x, y, 'ro')
plt.grid(True)

plt.figure()
plt.pcolormesh(gridx, gridy, grid)
plt.plot(x, y, 'ro')
plt.colorbar()

plt.show()