python - NumPy - 获取表示边界框的数组边界

Question

我有一个形状数组，(N, 4)以以下形式保存数据：

[[min_x, min_y, max_x, max_y],
 [min_x, min_y, max_x, max_y],
 ...,
 [min_x, min_y, max_x, max_y]]

我需要生成一个形状数组，(4,)这样：

[smallest_min_x, smallest_min_y, largest_max_x, largest_max_y]

标签： pythonnumpy

Answer 1

如果您的数组是arr，您可以使用：

np.concatenate((np.amin(arr, 0)[:2], np.amax(arr, 0)[2:]))

这将计算沿轴的最小值和最大值，选择所需的值并将它们连接起来。