numpy（python3）中奇怪数组的极值

Question

我有一个如下所示的数组：

ar=[[[678,701]],
 [[680,702]],
 [[674,710]],
 ...]

我必须找到每一列的极值（即，对于这些 678,680,674... 和 701,702,710,... 独立）。

我尝试通过以下方式访问这些列：

ar[:][0][0] 或 ar[:][0][1]

但事实证明，Python 对 ar[:][0] 的理解与 ar[0] 相同，因此我不知道有什么方法可以防止使用循环。是否还有任何复杂的技术可以做到这一点？

Answer 1

要找到沿特定轴的极值，您可以使用the axis parameter：

import numpy as np

ar = np.array([[[678,701]],
 [[680,702]],
 [[674,710]],
])

print ar.max(axis=0)  #  [[680 710]]
print ar.max(axis=0)  #  [[674 701]]

要切出两列，您可以使用：

print ar[:,0,0]       # [678 680 674]
print ar[:,0,1]       # [701 702 710]