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

要沿特定軸查找極值，可以使用axis參數 ：

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]