0

我有两个类似于这些的numpy数组，它们代表坐标：

import numpy as np
x=np.array([1,3,2,4,6,5,4,1])
y=np.array([4,4,3,2,2,1,3,5])

我也有n正方形：

s1 -> x=0 to 3, y=0 to 3
s2 -> x=3 to 6, y=3 to 6
s3 -> ...
s4 -> ...

我想计算每个正方形内的点数。这归结为评估n不平等。

我的方法很冗长并且（可能）效率低下：

count1=0
count2=0
count3=0
count4=0
for j in range(0, len(x)):
    #Square 1
    if x[j]<=3 and y[j]<=3:
        count1+=1
    #Square 2
    if x[j]<=3 and y[j]>3 and y[j]<=6:
        count2+=1
    #Square 3
    if x[j]>3 and x[j]<=6 and y[j]<=3:
        count3+=1
    #Square 4
    if x[j]>3 and x[j]<=6 and y[j]>3 and y[j]<=6:
        count4+=1

给定我的两个数组，这将返回：

In[1]: count1, count2, count3, count4
Out[1]: (1, 3, 4, 0)

我真正的问题包括可变数量的正方形（可能是 6，也可能是 36，等等）。

有没有一种方法可以自动生成count变量以及if语句的数量和边界？

4

2 回答 2

1

你没有列出你的整个代码，所以不清楚你到底想做什么。在任何情况下，您都可以用一个元组来描述每个正方形

square_n = ((x1, x2), (y1, y2))

并将它们放入字典中，其中键是这个元组，值是计数。然后，像

for square in squares_dict:
    (x1, x2), (y1, y2) = square
    if x1<a<x2 and y1<b<y2: # whatever criterion you have
        squares_dict[square] += 1
于 2018-08-30T15:10:42.663 回答
0

这是一个数组非常有用的情况。

countn您可以创建一个计数数组，并对该数组进行索引（因此count0变为count[0]、count1tocount[1]等），而不是创建单个变量。

现在，诀窍就是将 x 和 y 坐标映射到特定的计数数组索引，您只需进行一些数学运算即可。即，如果正方形是 3x3 并排列在一个大矩形中，则为x // 3 + (y // 3) * num_squares_per_row您提供索引。

如果您尝试计算的区域不统一，因此您无法提出一个简单的数学方程式，您始终可以制作一个字典，将您想要计算的内容映射到它们在count数组中的索引，并使用它来确定给定特定输入要增加的索引。

于 2018-08-30T15:06:23.553 回答