import numpy as np chunk_means = np.array([10., 20.]) counts = np.array([2, 6]) print(float(np.average(chunk_means, weights=counts))) print(float(chunk_means.mean())) print(int(counts.sum()))