以下是一个示例代码,用于按组计算几何平均值:
import numpy as np
def group_geometric_mean(data, groups):
result = []
for group in groups:
# 根据组别筛选数据
group_data = [d for d, g in zip(data, groups) if g == group]
# 计算几何平均值
geometric_mean = np.prod(group_data) ** (1 / len(group_data))
result.append(geometric_mean)
return result
# 示例数据
data = [2, 4, 6, 8, 10, 3, 5, 7, 9, 11]
groups = [1, 1, 1, 1, 1, 2, 2, 2, 2, 2]
# 调用函数计算几何平均值
result = group_geometric_mean(data, groups)
print(result)
在这个示例中,我们定义了一个group_geometric_mean
函数,它接受一个数据列表和一个组别列表作为参数。函数会按照组别筛选数据,并计算每个组别的几何平均值。最后,函数返回一个包含每个组别几何平均值的列表。
在示例数据中,我们有10个数据点,分为两个组别(组别1和组别2)。我们调用group_geometric_mean
函数,并打印结果。输出结果为两个几何平均值,分别对应于两个组别的数据。
上一篇:按组计算后包含缺失值的列数
下一篇:按组计算均值,排除某些行