import sys import os.path def compute_dist(filename): f = open(filename) users = {} for line in f: values = line.strip().split() users[values[0]] = int(values[2]) print sum(users.itervalues()) / float(len(users)) of = open(os.path.splitext(filename)[0] + "_dist.txt", "w") for degree in users.itervalues(): of.write(str(degree) + "\n") if __name__ == "__main__": compute_dist(sys.argv[1])