How To Find Max Element In Array In Python

In this tutorial, we will learn how to find the maximum element in an array in Python. Arrays are a fundamental data structure used to store and manipulate data.

Finding the maximum element is a common operation when working with arrays. There are several methods we can use to find the max element in an array, and they include using the built-in Python function max(), a loop, or the NumPy library.

We will explore each of these methods in this tutorial.

Method 1: Using the built-in max() function

The simplest and most efficient method is to use the built-in max() function. The max() function takes an iterable (such as a list, tuple, or array) as an argument and returns the maximum element. Here’s a simple example:

1
2
3
arr = [3, 9, 1, 7, 5]
max_element = max(arr)
print("Max element:", max_element)

Output

Max element: 9

Method 2: Using a loop

Another method to find the maximum element in an array is to use a loop. We can iterate through the array of elements and compare each element with the current maximum element. If the current array element is greater than the current maximum element, we update the maximum element value. Here’s an example:

1
2
3
4
5
6
7
8
arr = [3, 9, 1, 7, 5]
max_element = arr[0]
 
for num in arr:
    if num > max_element:
        max_element = num
 
print("Max element:", max_element)

Output

Max element: 9

Method 3: Using the NumPy library

NumPy is a powerful numerical computing library in Python that provides many useful functions for array manipulation. We can use the numpy.amax() function to find the maximum element in a NumPy array. First, we need to import the NumPy library and create a NumPy array:

1
2
3
4
5
import numpy as np
 
arr = np.array([3, 9, 1, 7, 5])
max_element = np.amax(arr)
print("Max element:", max_element)

Output

Max element: 9

This method is particularly efficient when working with large arrays.

Full Code

Here’s the full code for all three methods:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
## Method 1: Using max() function
arr = [3, 9, 1, 7, 5]
max_element = max(arr)
print("Max element (Method 1):", max_element)
 
## Method 2: Using a loop
max_element = arr[0]
 
for num in arr:
    if num > max_element:
        max_element = num
 
print("Max element (Method 2):", max_element)
 
## Method 3: Using NumPy library
import numpy as np
 
arr_np = np.array([3, 9, 1, 7, 5])
max_element = np.amax(arr_np)
print("Max element (Method 3):", max_element)
 
Max element (Method 1): 9
Max element (Method 2): 9
Max element (Method 3): 9

Conclusion

In this tutorial, we explored three methods to find the maximum element in an array in Python. Using the built-in max() function is the simplest and most efficient method for small arrays.

However, the loop and NumPy methods also provide ways to find the maximum element, with the latter being particularly efficient for large arrays.

Choose the method that best suits your specific requirements and become more proficient in array manipulation in Python.