{"id":475,"date":"2020-12-21T22:13:36","date_gmt":"2020-12-22T03:13:36","guid":{"rendered":"https:\/\/www.alpharithms.com\/?p=475"},"modified":"2022-06-16T15:23:57","modified_gmt":"2022-06-16T19:23:57","slug":"python-get-mean-average","status":"publish","type":"post","link":"https:\/\/www.alpharithms.com\/python-get-mean-average-361322\/","title":{"rendered":"Using Python to Get the Mean (Average) of Numbers"},"content":{"rendered":"

Python is arguably the most popular programming language for data science. As one might expect, it comes with a slew of built-in libraries that can handle statistical analysis such as mean, median, and mode calculations.<\/p>\n

Depending on your use\u2014there are several ways to approach using Python to calculate the average value of a set of numbers. Whether you’re in need of a weighed average, the harmonic mean, or something more exotic\u2014python has several average functions that are anything but!<\/p>\n

Obligatory Clarification<\/h2>\n

The term “average” and even “mean” are ambiguous and may refer to one (or several) different types of mathematical calculation. The\u00a0arithmetic<\/em> mean is the most common method\u2014and the one I’d wager most people\u00a0mean<\/em> when they broach the subject.<\/p>\n

Other types include the geometric mean, harmonic mean, and more than a dozen others such as the moving average. For a deeper dive into exactly\u00a0how some of these are calculated, check out this article<\/a>. For our discussion here, I’m going to glaze over the finer details and focus mostly on implementation in Python.<\/p>\n

Using Python to Get the Average<\/h2>\n

Below are several approaches to getting the mean in Python ranging from the simple mean function built into the statistics module to third-party libraries like numpy. For the examples, I’ll be using the following randomly-generated set of numbers:<\/p>\n

import random\r\n\r\n# Generate a list of 10 random numbers from 0-99 \r\nnumbers = [random.choice(range(100)) for _ in range(10)]\r\n\r\n>>> [59, 97, 94, 98, 54, 40, 96, 37, 11, 17]<\/pre>\n
These are the numbers that will be used for each example to follow.<\/p>\n
Statistics Library<\/h3>\n
Since Python 3.4 there is a standard statistics library<\/a> that provides several methods to calculate the mean of a set of numbers. Among them are methods to calculate arithmetic, geometric, and harmonic means as shown below:<\/p>\n
import statistics\r\n\r\n# Define a list of random numbers\r\nnumbers = [59, 97, 94, 98, 54, 40, 96, 37, 11, 17]\r\n\r\narithmetic_mean = statistics.mean(numbers)\r\n>>> 60.3\r\n\r\ngeometric_mean = statistics.geometric_mean(numbers)\r\n>>> 48.73877382924253\r\n\r\nharmonic_mean = statistics.harmonic_mean(numbers)\r\n>>> 35.868566290602814<\/pre>\n
Note<\/strong>: Before 3.4 Python provided similar functionality via the stats<\/code> library.<\/p>\n
Numpy<\/h3>\n
NumPy is among the most-used numerical processing libraries among data scientists, along with other staples such as pandas<\/code>, matplotlib<\/a>, and scikit-learn<\/code>. All of these are overkill for simple mean calculation but, if they’re already dependencies they can be convenient.<\/p>\n
import numpy\r\n\r\n# The arithmetic mean\r\nnumpy_mean = numpy.mean(numbers)\r\n>>> 60.3\r\n\r\n# The weighted average (without weights)\r\nnumpy_average = numpy.average(numbers)\r\n>>> 60.3<\/pre>\n
Note<\/strong>: The numpy.average<\/code> calculates a weighted average but, in the example above, isn’t provided with any data for weights. As such, a non-weighted average is calculated instead.<\/p>\n
SciPy.stats<\/h3>\n
The SciPy library is focused mostly on probability distributions but provides some functions for mean calculation. The arithmetic mean, along with median and mode functions, are available as attributes of other larger functions. Overall, the scipy.stats module isn’t used for simple mean calculation. If you’re hellbent on doing so, the geometric and harmonic means are available as such:<\/p>\n
from scipy import stats\r\n\r\n# The geometric mean\r\ngeometric_mean = stats.gmean(numbers)\r\n>>> 48.738773829242575\r\n\r\n# The harmonic mean\r\nharmonic_mean = stats.hmean(numbers)\r\n>>> 35.868566290602814<\/pre>\n
Manual Calculation<\/h3>\n
For those that prefer vanilla Python code\u2014the mean\/average isn’t exactly a hat trick. The following illustrates approaches for calculating the mean using nothing but standard Python syntax:<\/p>\n
# Arithmetic Mean\r\narithmetic_mean = sum(numbers) \/ len(numbers)\r\n>>> 60.3<\/pre>\n
The geometric mean and harmonic mean\u00a0can<\/em> be done in vanilla Python but they’re not nearly as straight-forward. At the very least, one would want to make use of the math.log<\/code> and math.pow<\/code> functions.<\/p>\n
Final Thoughts<\/h2>\n
Python makes it super easy to calculate the mean\u2014the toughest part is deciding which<\/em> method and\/or library to use to do so! The three Pythagorean means<\/a> are available as methods via the standard statistics<\/code> library. Other third-party libraries like numpy<\/code> can offer the same features but often add unwanted project overhead.<\/p>\n","protected":false},"excerpt":{"rendered":"
Python is arguably the most popular programming language for data science. As one might expect, it comes with a slew of built-in libraries that can handle statistical analysis such as mean, median, and mode calculations. Depending on your use\u2014there are several ways to approach using Python to calculate the average value of a set of […]<\/p>\n","protected":false},"author":2,"featured_media":476,"comment_status":"closed","ping_status":"closed","sticky":false,"template":"","format":"standard","meta":{"_acf_changed":false,"inline_featured_image":false,"footnotes":""},"categories":[655,10],"tags":[198,143,200,199,142],"acf":[],"_links":{"self":[{"href":"https:\/\/www.alpharithms.com\/wp-json\/wp\/v2\/posts\/475"}],"collection":[{"href":"https:\/\/www.alpharithms.com\/wp-json\/wp\/v2\/posts"}],"about":[{"href":"https:\/\/www.alpharithms.com\/wp-json\/wp\/v2\/types\/post"}],"author":[{"embeddable":true,"href":"https:\/\/www.alpharithms.com\/wp-json\/wp\/v2\/users\/2"}],"replies":[{"embeddable":true,"href":"https:\/\/www.alpharithms.com\/wp-json\/wp\/v2\/comments?post=475"}],"version-history":[{"count":3,"href":"https:\/\/www.alpharithms.com\/wp-json\/wp\/v2\/posts\/475\/revisions"}],"predecessor-version":[{"id":479,"href":"https:\/\/www.alpharithms.com\/wp-json\/wp\/v2\/posts\/475\/revisions\/479"}],"wp:featuredmedia":[{"embeddable":true,"href":"https:\/\/www.alpharithms.com\/wp-json\/wp\/v2\/media\/476"}],"wp:attachment":[{"href":"https:\/\/www.alpharithms.com\/wp-json\/wp\/v2\/media?parent=475"}],"wp:term":[{"taxonomy":"category","embeddable":true,"href":"https:\/\/www.alpharithms.com\/wp-json\/wp\/v2\/categories?post=475"},{"taxonomy":"post_tag","embeddable":true,"href":"https:\/\/www.alpharithms.com\/wp-json\/wp\/v2\/tags?post=475"}],"curies":[{"name":"wp","href":"https:\/\/api.w.org\/{rel}","templated":true}]}}