我正在嘗試對包含 numpy 數(shù)組的列表進(jìn)行一些計(jì)算（平均值、總和等）。例如：列表 = [數(shù)組([2, 3, 4]),數(shù)組([4, 4, 4]),數(shù)組([6, 5, 4])]如何檢索平均值（例如）？在類似列表[4,4,4]或類似 numpy 的數(shù)組中array([4,4,4])？在此先感謝您的幫助！編輯：抱歉，我沒有正確解釋我的目標(biāo)：我想獲得數(shù)組的第 i 個索引的平均值。例如，對于索引 0 ：(2+4+6)/3 = 4我不想要這個：(2+3+4)/3 = 3因此最終的結(jié)果將是[4,4,4] / 而不是 [3,4,5]
3 回答
哆啦的時光機(jī)
TA貢獻(xiàn)1779條經(jīng)驗(yàn) 獲得超6個贊
如果L是標(biāo)量列表，則可以使用直接表達(dá)式來計(jì)算平均值：
sum(L) / len(L)
幸運(yùn)的是，這在數(shù)組列表上沒有改變：
L = [np.array([2, 3, 4]), np.array([4, 4, 4]), np.array([6, 5, 4])]
sum(L) / len(L)
# array([4., 4., 4.])
對于這個例子，這恰好比 numpy 函數(shù)快得多 np.mean
timeit(lambda: np.mean(L, axis=0))
# 13.708808058872819
timeit(lambda: sum(L) / len(L))
# 3.4780975924804807
Cats萌萌
TA貢獻(xiàn)1805條經(jīng)驗(yàn) 獲得超9個贊
給定一個 1d array a，np.mean(a)應(yīng)該可以解決問題。
如果您有一個 2d 數(shù)組并希望分別計(jì)算每個數(shù)組的均值，請指定np.mean(a, axis=1).
有等價的函數(shù)np.sum，等等。
https://docs.scipy.org/doc/numpy/reference/generated/numpy.mean.html
https://docs.scipy.org/doc/numpy/reference/generated/numpy.sum.html
搖曳的薔薇
TA貢獻(xiàn)1793條經(jīng)驗(yàn) 獲得超6個贊
您可以使用 map
import numpy as np
my_list = [np.array([2, 3, 4]),np.array([4, 4, 4]),np.array([6, 5, 4])]
np.mean(my_list,axis=0) #[4,4,4]
注意：不要命名您的變量，list因?yàn)樗鼤绊憙?nèi)置函數(shù)
添加回答
舉報
0/150
提交
取消