How do I calculate the variance of a NumPy array in Python?

How do I calculate the variance of a NumPy array in Python?

You can calculate the variance of a NumPy array using the numpy.var() function. Just pass your array as an argument.

Can you show me an example?

Sure! Here's a simple example:

import numpy as np

data = np.array([1, 2, 3, 4, 5])
variance = np.var(data)
print(variance)

This will give you the variance of the array.

What if I want the sample variance instead of the population variance?

For sample variance, you can set the ddof parameter to 1 in the numpy.var() function:

sample_variance = np.var(data, ddof=1)

Thanks! Is there a way to calculate variance along a specific axis?

Yes, you can specify the axis parameter in the numpy.var() function to calculate variance along a specific axis.