NumPy: Create a new vector with 2 consecutive 0 between two values of a given vector

NumPy: Array Object Exercise-185 with Solution

Insert zeros between elements in an array.

Write a NumPy program to create a new vector with 2 consecutive 0 between two values of a given vector.

Pictorial Presentation:

Sample Solution:

Python Code:

# Importing the NumPy library
import numpy as np

# Creating a NumPy array 'nums' containing numbers from 1 to 8
nums = np.array([1, 2, 3, 4, 5, 6, 7, 8])

# Displaying the original array 'nums'
print("Original array:")
print(nums)

# Variable 'p' set to 2
p = 2

# Creating a new array 'new_nums' of length len(nums) + (len(nums) - 1) * p filled with zeros
new_nums = np.zeros(len(nums) + (len(nums) - 1) * (p))

# Filling the 'new_nums' array with elements from 'nums' at intervals of (p + 1)
new_nums[::p + 1] = nums

# Displaying the new array 'new_nums'
print("\nNew array:")
print(new_nums)

Sample Output:

Original array:
[1 2 3 4 5 6 7 8]

New array:
[1. 0. 0. 2. 0. 0. 3. 0. 0. 4. 0. 0. 5. 0. 0. 6. 0. 0. 7. 0. 0. 8.]

Explanation:

In the above code -

nums = np.array([1,2,3,4,5,6,7,8]): Initializes a NumPy array nums with values 1 to 8.

p = 2: Sets the value of p to 2, meaning 2 zeros will be inserted between each element of the original array.

new_nums = np.zeros(len(nums) + (len(nums)-1)*(p)): This line creates a new NumPy array new_nums filled with zeros. The length of new_nums is calculated based on the length of the original array and the number of zeros to be inserted between each element.

new_nums[::p+1] = nums: This line assigns the elements of the original array nums to the corresponding positions in the new array new_nums, skipping p+1 indices for each assignment.

Python-Numpy Code Editor:

Previous:Write a NumPy program to create an array using generator function that generates 15 integers.
Next: Write a Numpy program to multiply an array of dimension (2,2,3) by an array with dimensions (2,2).