w3resource

NumPy: Broadcast on different shapes of arrays where p(3,3) + q(3)

NumPy: Array Object Exercise-124 with Solution

Write a NumPy program to broadcast on different shapes of arrays where p(3,3) + q(3).

The p variable has a shape of (3, 3), while q only has a shape of 3.

Pictorial Presentation:

NumPy: Broadcast on different shapes of arrays where p(3,3) + q(3)

Sample Solution:

Python Code:

import numpy as np    
p = np.array([[0, 0, 0],
              [1, 2, 3],
              [4, 5, 6]]) 
q= np.array([10, 11, 12]) 
print("Original arrays:")
print("Array-1")
print(p)
print("Array-2")
print(q)
print("\nNew Array:")
new_array1 = p + q 
print(new_array1)

Sample Output:

Original arrays:
Array-1
[[0 0 0]
 [1 2 3]
 [4 5 6]]
Array-2
[10 11 12]

New Array:
[[10 11 12]
 [11 13 15]
 [14 16 18]]

Python Code Editor:

Have another way to solve this solution? Contribute your code (and comments) through Disqus.

Previous: Write a NumPy program to create two arrays when the size is bigger or smaller than a given array
Next: Write a NumPy program to broadcast on different shapes of arrays where a(,3) + b(3).

What is the difficulty level of this exercise?

Test your Python skills with w3resource's quiz



Python: Tips of the Day

Set comprehension:

>>> m = {x ** 2 for x in range(5)}
>>> m
{0, 1, 4, 9, 16}