Question

到目前为止，我一直在将数组与单个变量进行比较，例如球员位置：

for position in ships:
    if player_location_x > position[0]-100 and player_location_x < position[0]+100 and player_location_y > position[1]-100 and player_location_y < position[1]+100:
        #Do something (e.g. draw bullets between the two locations)

我如何将其扩展为比较数组本身的值，例如比较'船'的x和y值来检查它们之间的距离，还比较'船'和'more_ships'之间的x和y值？

ships = numpy.array([
                                [
                                  shuttle_class.npc_x[0],  
                                  shuttle_class.npc_y[0],  
                                  shuttle_class.img,    
                                  shuttle_class.hp
                                ],

                                [
                                  shuttle_class.npc_x[1],  
                                  shuttle_class.npc_y[1],  
                                  shuttle_class.img,    
                                  shuttle_class.hp
                                ],

                                [
                                  shuttle_class.npc_x[2],  
                                  shuttle_class.npc_y[2],  
                                  shuttle_class.img,    
                                  shuttle_class.hp
                                ]
                    ])

more_ships = numpy.array([
                                [
                                  shuttle_class.npc_x[3],  
                                  shuttle_class.npc_y[3],  
                                  shuttle_class.img,    
                                  shuttle_class.hp
                                ],

                                [
                                  shuttle_class.npc_x[4],  
                                  shuttle_class.npc_y[4],  
                                  shuttle_class.img,    
                                  shuttle_class.hp
                                ],

                                [
                                  shuttle_class.npc_x[5],  
                                  shuttle_class.npc_y[5],  
                                  shuttle_class.img,    
                                  shuttle_class.hp
                                ]
                         ])

Answer 1

让我们从船只的两个数组x1, y1开始。您希望使用x2, y2生成每个成对距离。让我们为讨论说你有5艘船和3艘船。所以我们使用numpy meshgrid：

xx1, xx2 = np.meshgrid(x1, x2)  # both these are 3x5 arrays
yy1, yy2 = np.meshgrid(y1, y2)
dx = xx1 - xx2 # np.abs(xx1 - xx2) if you want just absolute values
dy = yy1 - yy2

现在，您可以使用np.where获取最终列表：

sel = np.where( (dx <= d_max) & (dy <= d_max) )

sel是一个2xn数组。这些值是满足条件的n点的索引。

编辑：根据OP的要求添加示例代码。

import matplotlib.pyplot as plt
import numpy as np

sx = np.array([0, 250, 500])
sy = np.array([100, 100, 100])
mx = np.array([1000, 0])
my = np.array([0,0])
plt.scatter(sx, sy)
plt.scatter(mx, my, c='r', marker='o')
plt.grid()

我们有三艘船（s）和两艘more_ships（m）。

xx1, xx2 = np.meshgrid(sx, mx)
yy1, yy2 = np.meshgrid(sy, my)

让我们稍微研究一下：np.shape(xx1)是(2,3)。第一个索引是m（更多船只），第二个索引是s。

dx = np.abs(xx1 - xx2)
dy = np.abs(yy1 - yy2)
d_max = 200.0
sel = np.where( (dx <= d_max) & (dy <= d_max) )
print sel

你会看到sel有两个数组。第一个数组引用第一个轴的索引（m），第二个数组引用（s）。在这种情况下，数组的值为1和0，这意味着more_ships[1]在ships[0]的200像素范围内。

如果您将sx更改为np.array([0, 250, 1000])，那么sel将为array([0, 1]), array([2, 0]) - 这意味着more_ships[0]位于ships[2]的200像素范围内，并且more_ships[1]靠近ships[0]。

如何比较数组（不完全匹配）

1 个答案: