Last modified: August 03, 2024

This article is written in: 🇺🇸

Joining and Splitting Arrays

In NumPy, manipulating the structure of arrays is a common operation. Whether combining multiple arrays into one or splitting a single array into several parts, NumPy provides a set of intuitive functions to achieve these tasks efficiently. This guide will cover various methods to join and split arrays, providing detailed explanations and practical examples.

Stacking Arrays

Stacking is the technique of joining arrays along a new axis. The function for this is np.stack(), and the axis on which you want to stack the arrays can be specified using the axis parameter.

Example: Stacking Along a New Axis

import numpy as np

a = np.array([[1, 2], [3, 4]])
b = np.array([[5, 6], [7, 8]])

# Vertical stacking (default axis=0)
c = np.stack((a, b))
print("Vertically stacked:\n", c)

# Horizontal stacking (axis=1)
d = np.stack((a, b), axis=1)
print("\nHorizontally stacked:\n", d)

Expected output:

Vertically stacked:
[[[1 2]
  [3 4]]

 [[5 6]
  [7 8]]]

Horizontally stacked:
[[[1 2]
  [5 6]]

 [[3 4]
  [7 8]]]

• Vertical Stacking refers to the process of joining arrays along a new first axis, resulting in a higher-dimensional array. This technique is often used to stack 1D arrays into a 2D array or 2D arrays into a 3D array, effectively increasing the number of dimensions.
• When Horizontal Stacking is performed, arrays are joined along a new second axis. This method arranges elements differently, typically placing them side by side in a new array. It is commonly used to concatenate 1D arrays into a larger 1D array or 2D arrays into a wider 2D array, without increasing the number of dimensions.

Concatenating Arrays

Concatenation merges arrays along an existing axis. Use np.concatenate() for this.

Example: Concatenation Along Existing Axes

a = np.array([[1, 2], [3, 4]])
b = np.array([[5, 6], [7, 8]])

# Vertical concatenation (default axis=0)
c = np.concatenate((a, b))
print("Vertically concatenated:\n", c)

# Horizontal concatenation (axis=1)
d = np.concatenate((a, b), axis=1)
print("\nHorizontally concatenated:\n", d)

Expected output:

Vertically concatenated:
[[1 2]
 [3 4]
 [5 6]
 [7 8]]

Horizontally concatenated:
[[1 2 5 6]
 [3 4 7 8]]

• Vertical Concatenation involves adding new rows to an existing array, thus extending it in the vertical direction. This method increases the number of rows in the array while keeping the number of columns the same.
• Through Horizontal Concatenation, new columns are added to an existing array, extending it horizontally. This process increases the number of columns in the array, allowing for a broader set of data to be included alongside the existing columns.

Appending to Arrays

You can append values to an array using np.append(). It adds values at the end along the specified axis.

Example: Appending Values

a = np.array([1, 2, 3])
b = np.array([4, 5, 6])

# Append b to a
c = np.append(a, b)
print("Appended array:\n", c)

Expected output:

Appended array:
[1 2 3 4 5 6]

Appending Arrays involves combining the elements of the second array to the end of the first array, resulting in the creation of a new combined array. This operation effectively extends the first array by adding the elements from the second array in sequence.

Splitting Arrays

Splitting breaks down an array into smaller chunks. np.split() is the function used to divide arrays.

Regular and Custom Splits

a = np.array([1, 2, 3, 4, 5, 6])

# Split into three equal parts
b = np.split(a, 3)
print("Regular split:\n", b)

# Split at the 2nd and 4th indices
c = np.split(a, [2, 4])
print("\nCustom split:\n", c)

Expected output:

Regular split:
[array([1, 2]), array([3, 4]), array([5, 6])]

Custom split:
[array([1, 2]), array([3, 4]), array([5, 6])]

• Regular Split divides an array into equal parts. If the array cannot be evenly divided, an error is typically raised, ensuring that each resulting segment has the same number of elements.
• In a Custom Split, the array is divided at specified indices, providing more flexibility in how the array is split. This allows for creating segments of varying sizes according to the specified positions.

Advanced Joining and Splitting Techniques

Example: HStack and VStack

• Horizontal Stack (hstack) is used to combine arrays horizontally along columns. This operation aligns arrays side by side, effectively increasing the number of columns in the resulting array.
• With Vertical Stack (vstack), arrays are combined vertically along rows. This method stacks arrays on top of each other, increasing the number of rows in the final array.

a = np.array([1, 2, 3])
b = np.array([4, 5, 6])

# Horizontal stack
h_stack = np.hstack((a, b))
print("Horizontal stack:\n", h_stack)

# Vertical stack
v_stack = np.vstack((a, b))
print("\nVertical stack:\n", v_stack)

Expected output:

Horizontal stack:
[1 2 3 4 5 6]

Vertical stack:
[[1 2 3]
 [4 5 6]]

Example: DStack

• Depth Stack (dstack): Stacks arrays along the third axis (depth).

a = np.array([[1, 2], [3, 4]])
b = np.array([[5, 6], [7, 8]])

# Depth stack
d_stack = np.dstack((a, b))
print("Depth stack:\n", d_stack)

Expected output:

Depth stack:
[[[1 5]
  [2 6]]

 [[3 7]
  [4 8]]]

Practical Applications and Considerations

Understanding how to join and split arrays is crucial for various data manipulation tasks, including:

1. Data Preprocessing involves combining and splitting datasets to prepare them for training and testing machine learning models, ensuring that the data is properly formatted and ready for analysis.
2. Data Augmentation creates variations of existing data, which helps in training robust machine learning models by providing diverse examples for learning.
3. Feature Engineering includes the process of combining different feature sets into a single array, facilitating comprehensive analysis and enhancing the model's ability to learn relevant patterns.
4. Batch Processing refers to splitting large datasets into smaller, manageable batches, which allows for efficient processing and analysis, especially when dealing with limited computational resources.

Summary Table