我有以下 numpy 陣列
[[7, 0, 0, 6],
[5, 6, 6, 1],
[4, 1, 6, 7],
[5, 3, 4, 7]]
我想使用 np.max 找到每列中的最大值,然后在一個物件中列印出結果,這樣輸出將如下所示
[7, 6, 6, 7]
uj5u.com熱心網友回復:
如果arr是您的陣列,那么您只需要使用該max函式,指示所選軸:
arr.max(axis=0)
輸出:
array([7, 6, 6, 7])
如果你想要一個串列而不是一個 numpy 陣列:
arr.max(axis=0).tolist()
輸出:
[7, 6, 6, 7]
uj5u.com熱心網友回復:
您可以遍歷轉置陣列并使用以下命令查找最大值np.max():
import numpy as np
m =np.array([[7, 0, 0, 6],
[5, 6, 6, 1],
[4, 1, 6, 7],
[5, 3, 4, 7]])
out = [np.max(i) for i in m.transpose()]
print(out)
輸出:
[7, 6, 6, 7]
uj5u.com熱心網友回復:
import numpy as np
m =np.array([[7, 0, 0, 6],
[5, 6, 6, 1],
[4, 1, 6, 7],
[5, 3, 4, 7]])
#list
max_numbers = [max(x) for x in m]
#array
max_num_array = np.array(max_numbers)
轉載請註明出處,本文鏈接:https://www.uj5u.com/ruanti/497800.html
