A我有一個shape陣列(2,4,1)。我想計算 和 的平均值A[0]并將A[1]這兩個平均值存盤在A_mean. 我介紹了當前和預期的輸出。
import numpy as np
A=np.array([[[1.7],
[2.8],
[3.9],
[5.2]],
[[2.1],
[8.7],
[6.9],
[4.9]]])
for i in range(0,len(A)):
A_mean=np.mean(A[i])
print(A_mean)
當前輸出為
5.65
預期的輸出是
[3.4,5.65]
uj5u.com熱心網友回復：
for 回圈不是必需的，因為 NumPy 已經知道如何對向量/矩陣進行操作。
解決方案是洗掉回圈并按如下方式更改軸：
A_mean=np.mean(A, axis=1)
print(A_mean)
輸出：
[[3.4 ]
[5.65]]
現在您還可以進行一些編輯以洗掉括號[3.4 5.65]：
print(A_mean.ravel())
uj5u.com熱心網友回復：
嘗試這個。
import numpy as np
A=np.array([[[1.7],
[2.8],
[3.9],
[5.2]],
[[2.1],
[8.7],
[6.9],
[4.9]]])
A_mean = []
for i in range(0,len(A)):
A_mean.append(np.mean(A[i]))
print(A_mean)
轉載請註明出處，本文鏈接：https://www.uj5u.com/shujuku/507403.html