如何索引已附加到 1d numpy 数组 python 的元素

Question

我有一个问题，我一直在努力解决。我有一个 numpy 数组，我在末尾附加值（如下所示）：

array = []
for (x, y, w, h) in e:
    wm = int(x + float(w / 2))
    hm = int(y + float(h / 2)) 
    array.append([numpy.float32(wm),numpy.float32(wh)])
return numpy.array(array)

#and it gives 
[200.90489, 100.39489] #the second last 
[288.0594, 30.0950] #last
#and so on..

我想索引array[0][-2] - array[0][-1]，这样我就可以找到第一列中倒数第二个附加元素（200.90489）和第一列中最后一个附加元素（288.0594）之间的区别。 我可以索引列，但我不能索引行，因为它是一维数组，我尝试了多种方法将其转换为二维，但它不起作用。我试过 numpy.resize 和 numpy.reshape 但都没有帮助。
我该怎么办？ 提前致谢！

Answer 1

几个cmets：

• array 是 numpy 的一个关键，你可能不应该将它用作变量名。它会起作用，但这不是最佳做法。

• return 用于函数末尾的返回值，不能将其放入 for 循环中（另外，即使在函数中，循环也会在第一个 @987654324 处中断@)

• 看起来您的结果将是一个二维数组，array[i][j] 应该可以工作。但我认为你交换了i 和j。

• 不知道为什么要转换为 numpy.array，它已经适用于列表。

• 很多明显不必要的类型转换：float => int => numpy.float32。我会在下面假设你想要一个四舍五入的整数。

所以如果我猜对了你想要做什么，这应该可以工作：

my_array = []
for (x, y, w, h) in e:
    wm = int(x + w/2.)
    hm = int(y + h/2.) 
    my_array.append([wm, wh])

if len(my_array) >= 2:
    last_wm_diff = my_array[-1][0]-my_array[-2][0]
    last_hm_diff = my_array[-1][1]-my_array[-2][1]

# It will work the same if you convert to a numpy array:
my_array = numpy.array(my_array)

if my_array.shape[0] >= 2:
    last_wm_diff = my_array[-1][0]-my_array[-2][0]
    last_hm_diff = my_array[-1][1]-my_array[-2][1]

Answer 2

确保您的数组的形状为 (1,2)

那就试试吧：

np.append(array,[numpy.float32(wm),numpy.float32(wh)], axis=0)