PYTHON

Python NumPy insert Function Example Using insertion at different points

# welcome to softhunt.net
# Python Program illustrating
# numpy.insert()

import numpy as np

#Working on 1D
arr = np.arange(6).reshape(2, 3)
print("1D arr : 
", arr)
print("Shape : ", arr.shape)

# value = 9
# index = 1
# Insertion before first index
a = np.insert(arr, (2, 4), 9)
print("
Insertion at two points : ", a)
print("Shape : ", a.shape)


# Working on 2D array
arr = np.arange(12).reshape(3, 4)
print("

2D arr : 
", arr)
print("Shape : ", arr.shape)
a = np.insert(arr, (0, 3), 66, axis = 1)
print("
Insertion at two points : 
", a)
print("Shape : ", a.shape)