python - 数字对的一个索引的平均值

Question

我有一张像 [[2,3]、[7,6]、[12,567]、[18,4]] 这样的表格。我需要元素 2、7、12、18 的平均值。

def average(data):
    temp = []
    for x in data:
        temp.append(x[0])
    return np.average(temp)

有没有更好的办法？

标签： pythonnumpy

Answer 1

你可以使用：

from statistics import mean
mean(e[0] for e in data)

或者：

np.average([e[0] for e in data])

输出：

9.75

或者：

np.average(np.array(data)[:,0])

@WillemVanOnsem 的建议也很棒：

np.array(data)[:,0].mean()

---

def average(data):
    np.average([e[0] for e in data])