Python NumPy sum Function

🎓 Top 15 Udemy Courses (80-90% Discount): My Udemy Courses - Ramesh Fadatare — All my Udemy courses are real-time and project oriented courses.

▶️ Subscribe to My YouTube Channel (178K+ subscribers): Java Guides on YouTube

▶️ For AI, ChatGPT, Web, Tech, and Generative AI, subscribe to another channel: Ramesh Fadatare on YouTube

The sum function in Python's NumPy library is used to compute the sum of array elements over a specified axis. This function is essential in various fields such as data analysis, statistics, and scientific computing where summing operations are required.

Table of Contents

1. Introduction
2. Importing the numpy Module
3. sum Function Syntax
4. Understanding sum
5. Examples
   • Basic Usage
   • Computing Sum Along an Axis
   • Handling Special Values
6. Real-World Use Case
7. Conclusion
8. Reference

Introduction

The sum function in Python's NumPy library allows you to compute the sum of elements along a specified axis in an array. This function is particularly useful in numerical computations where summing operations are necessary.

Importing the numpy Module

Before using the sum function, you need to import the numpy module, which provides the array object.

import numpy as np

sum Function Syntax

The syntax for the sum function is as follows:

np.sum(a, axis=None, dtype=None, out=None, keepdims=<no value>)

Parameters:

• a: The input array containing elements whose sum is to be computed.
• axis: Optional. The axis along which to compute the sum. If not provided, the sum of all elements is computed.
• dtype: Optional. The data type of the output array.
• out: Optional. A location into which the result is stored.
• keepdims: Optional. If True, the axes which are reduced are left in the result as dimensions with size one.

Returns:

• An array with the sum of elements along the specified axis.

Understanding sum

The sum function computes the sum of elements along a specified axis in the input array. If the axis parameter is not provided, it computes the sum of all elements in the array.

Examples

Basic Usage

To demonstrate the basic usage of sum, we will compute the sum of all elements in an array.

Example

import numpy as np

# Array of values
values = np.array([1, 2, 3, 4])

# Computing the sum of all elements
total_sum = np.sum(values)
print(total_sum)

Output:

10

Computing Sum Along an Axis

This example demonstrates how to compute the sum of elements along a specified axis in a two-dimensional array.

Example

import numpy as np

# 2D array of values
values = np.array([[1, 2, 3], [4, 5, 6]])

# Computing the sum along axis 0 (columns)
sum_axis_0 = np.sum(values, axis=0)
print(sum_axis_0)

# Computing the sum along axis 1 (rows)
sum_axis_1 = np.sum(values, axis=1)
print(sum_axis_1)

Output:

[5 7 9]
[ 6 15]

Handling Special Values

This example demonstrates how sum handles special values such as NaNs and infinities.

Example

import numpy as np

# Array with special values
special_values = np.array([1, 2, np.nan, 4, np.inf])

# Computing the sum of all elements
special_sum = np.sum(special_values)
print(special_sum)

Output:

nan

You can handle these special values by using the np.nansum function to ignore NaNs during the summation.

Example

import numpy as np

# Array with special values
special_values = np.array([1, 2, np.nan, 4, np.inf])

# Computing the sum of all elements, ignoring NaNs
special_sum = np.nansum(special_values)
print(special_sum)

Output:

inf

Real-World Use Case

Data Analysis

In data analysis and reporting, the sum function can be used to compute the total of numerical data, such as calculating the total sales or total number of items.

Example

import numpy as np

def total_sales(sales_data):
    return np.sum(sales_data)

# Example usage
sales_data = np.array([150.75, 200.50, 100.25, 175.80])
total = total_sales(sales_data)
print(f"Total Sales: {total}")

Output:

Total Sales: 627.3

Conclusion

The sum function in Python's NumPy library is used for computing the sum of elements along a specified axis in an array. This function is useful in various numerical and data processing applications, particularly those involving summing operations. Proper usage of this function can enhance the accuracy and efficiency of your computations.

Reference

Python NumPy sum Function