Python numpy append() Function Usage

Python numpy append() function is used to merge two arrays. This function returns a new array and the original array remains unchanged.

numpy.append() Syntax

The function syntax is:

numpy.append(arr, values, axis=None)

The arr can be an array-like object or a NumPy array. The values are appended to a copy of this array.
The values are array-like objects and it’s appended to the end of the “arr” elements.
The axis specifies the axis along which values are appended. If the axis is not provided, both the arrays are flattened.

Python numpy.append() Examples

1. Flattening Two Arrays

import numpy as np

arr1 = np.array([[1, 2], [3, 4]])
arr2 = np.array([[10, 20], [30, 40]])

# no axis provided, array elements will be flattened
arr_flat = np.append(arr1, arr2)

print(arr_flat)  # [ 1  2  3  4 10 20 30 40]

2. Merging Along Axis

import numpy as np

arr_merged = np.append([[1, 2], [3, 4]], [[10, 20], [30, 40]], axis=0)
print(f'Merged 2x2 Arrays along Axis-0:\n{arr_merged}')

arr_merged = np.append([[1, 2], [3, 4]], [[10, 20], [30, 40]], axis=1)
print(f'Merged 2x2 Arrays along Axis-1:\n{arr_merged}')

Output:

Merged 2x2 Arrays along Axis-0:
[[ 1  2]
 [ 3  4]
 [10 20]
 [30 40]]
Merged 2x2 Arrays along Axis-1:
[[ 1  2 10 20]
 [ 3  4 30 40]]

When the 2×2 arrays are merged along the x-axis, the output array size is 4×2.
When the 2×2 arrays are merged along the y-axis, the output array size is 2×4.

3. numpy.append() – Merging Arrays of different shapes

The append() function throws ValueError if both the arrays are of different shape, excluding the axis. Let’s understand this scenario with a simple example.

arr3 = np.append([[1, 2]], [[1, 2, 3], [1, 2, 3]])
print(arr3)

arr3 = np.append([[1, 2]], [[1, 2], [3, 4]], axis=0)
print(arr3)

In the first example, the array elements are flattened. So even if they have completely different size – 1×2 and 2×3, the append() works fine.
In the second example, the array shapes are 1×2 and 2×2. Since we are appending along the 0-axis, the 0-axis shape can be different. The other shapes should be the same, so this append() will also work fine.

Output:

[1 2 1 2 3 1 2 3]

[[1 2]
 [1 2]
 [3 4]]

Let’s look at another example where ValueError will be raised.

import numpy as np

arr3 = np.append([[1, 2]], [[1, 2, 3]], axis=0)
Traceback (most recent call last):
  File "", line 1, in 
  File "/Library/Frameworks/Python.framework/Versions/3.7/lib/python3.7/site-packages/numpy/lib/function_base.py", line 4528, in append
    return concatenate((arr, values), axis=axis)
ValueError: all the input array dimensions except for the concatenation axis must match exactly

 

Python numpy append() Function Usage

Create a Free Account

Register now and get access to our Cloud Services.

Posts you might be interested in: