我有 numpy 陣列，例如 np.array([2,2])
[[1,9],
[7,3]]
我想獲得第三個維度的最大值并將其變為一維。
那么 numpy.array 應該是這樣的 [9,7]
我想我可以用 for 回圈來做這個，然后再做一個 numpy.
但是它看起來很確認，有什么好的方法可以做到這一點嗎？
uj5u.com熱心網友回復：
amax函式（別名是np.max）
import numpy as np
a = np.array([[1,9],
[7,3]])
np.amax(a, axis=1)
# array([9, 7])
uj5u.com熱心網友回復：
將 max 與特定軸一起使用。在此示例中，軸為 1。
import numpy as np
arr = np.array([[1,9],
[7,3]])
arr_max = np.max(arr, axis=1)
print(arr_max)
輸出：
[9 7]
uj5u.com熱心網友回復：
numpy.max只是. _ _ numpy.amax此函式僅適用于單個輸入陣列，并在整個陣列中查找最大元素的值（回傳一個標量）。或者，它接受一個軸引數，并將沿著輸入陣列的軸找到最大值（回傳一個新陣列）。
import numpy
np_array = numpy.array([[1,9],
[7,3]])
max_array = numpy.max(np_array, axis=1)
print(max_array.shape)
print(max_array)
輸出：
(2,)
[9 7]
轉載請註明出處，本文鏈接：https://www.uj5u.com/houduan/516043.html
標籤：Python麻木的