w3resource

NumPy: Broadcast on different shapes of arrays where a(,3) + b(3)

NumPy: Array Object Exercise-125 with Solution

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

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

Pictorial Presentation:

NumPy: Broadcast on different shapes of arrays where a(,3) + b(3)

Sample Solution:

Python Code:

import numpy as np    
p = np.array([[0], [10], [20]]) 
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]
 [10]
 [20]]
Array-2
[10 11 12]

New Array:
[[10 11 12]
 [20 21 22]
 [30 31 32]]

Python Code Editor:

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

Previous: Write a NumPy program to broadcast on different shapes of arrays where p(3,3) + q(3).
Next:Write a NumPy program to rearrange the dimensions of a given array.

What is the difficulty level of this exercise?

Test your Python skills with w3resource's quiz



Python: Tips of the Day

List comprehension:

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