numpy np数组在列表中计算平均值

zf2sa74q 于 2023-10-19 发布在其他

关注(0)|答案(2)|浏览(158)

我试图计算每个嵌套列表的平均值：

import numpy as np
centroids = [[[3, 2]], [[2, 3], [3, 4], [2, 3]], [[1, 2], [3, 4]]]

所以我希望这些列的平均值如下：[3, 2]、[[2, 3], [3, 4], [2, 3]] = [2.3, 3.3]和[[1, 2], [3, 4]] = [2, 3]
我试着去做：

for c in centroids:
    c = np.mean(np.array(c), axis=0)

但它不起作用，质心列表中没有任何变化。怎么做？

numpy

来源：https://stackoverflow.com/questions/76913608/np-array-inside-list-to-compute-mean

2条答案

按热度按时间

eqoofvh91#

使用简单的列表理解来收集方法：

centroids = [np.mean(a, 0).tolist() for a in centroids]

[[3.0, 2.0], [2.3333333333333335, 3.3333333333333335], [2.0, 3.0]]

赞(0）回复(0）举报 2023-10-19

n3h0vuf22#

一个vanilla python实现：

centroids = [[[3, 2]], [[2, 3], [3, 4], [2, 3]], [[1, 2], [3, 4]]]
means = list()
for centroid in centroids:
    pair = [0, 0]
    for centroidPair in centroid:
        pair[0] += centroidPair[0]
        pair[1] += centroidPair[1]
    pair[0] /= len(centroid)
    pair[1] /= len(centroid)
    means.append(pair);
print(means);

它输出预期结果：

[[3.0, 2.0], [2.3333333333333335, 3.3333333333333335], [2.0, 3.0]]

赞(0）回复(0）举报 2023-10-19

我来回答

numpy np数组在列表中计算平均值

2条答案

相关问题

热门标签

最新问答