How to split a 2D array in Numpy

Overview

In Python, an array is a data structure used to store multiple items of the same type. Arrays are useful when dealing with many values of the same data type. An array needs to explicitly import the array module for declaration.

A 2D array is simply an array of arrays. The numpy.array_split() method in Python is used to split a 2D array into multiple sub-arrays of equal size.

Parameters

array_split() takes the following parameters:

  • array (required): Represents the input array.
  • indices_or_section (required): Represents the number of splits to be returned.
  • axis (optional): The axis along which the values are appended.

To split a 2D array, pass in the array and specify the number of splits you want.

Example

Now, let’s split a 2D array into three sections or indices.

Code

import numpy as np
array1 = np.array([[1, 2], [3, 4], [5, 6], [7, 8], [9, 10]])
# splitting the array into three indexes
new_array1 = np.array_split(array1, 3)
print('The arrays splitted into 3 sections are:', new_array1)

Explanation

  • Line 1: We import the numpy module.
  • Line 3: We create an array array1.
  • Line 5: We use the numpy.array_split() method to split the given array into three sections, and then assign the result to a variable named new_array1.
  • Line 6: We print the split 2D array, new_array1.

Free Resources

License: Creative Commons-Attribution-ShareAlike 4.0 (CC-BY-SA 4.0)