Extracting the real and imaginary parts of an NumPy array of complex numbers

Last Updated : 02 Sep, 2020

Numpy library gives us functions such as real() and imag() to find real and imaginary parts of a complex number.

real() : To find real part of the complex number
imag() : To find imaginary part of the complex number

Example 1 :

# importing the module 
import numpy as np 
  
# creating a NumPy array 
complex_num = np.array([-1 + 9j, 2 - 77j, 
                        31 - 25j, 40 - 311j, 
                        72 + 11j]) 
  
# traversing the list 
for i in range(len(complex_num)): 
      print("{}. complex number is {}".format(i + 1, complex_num[i])) 
      print ("The real part is: {}".format(complex_num[i].real)) 
      print ("The imaginary part is: {}\n".format(complex_num[i].imag)) 

Output :

1. complex number is (-1+9j)
The real part is: -1.0
The imaginary part is: 9.0

2. complex number is (2-77j)
The real part is: 2.0
The imaginary part is: -77.0

3. complex number is (31-25j)
The real part is: 31.0
The imaginary part is: -25.0

4. complex number is (40-311j)
The real part is: 40.0
The imaginary part is: -311.0

5. complex number is (72+11j)
The real part is: 72.0
The imaginary part is: 11.0

Example 2 : The imaginary value of real number will be 0.

# importing the module 
import numpy as np 
  
# creating a NumPy array 
complex_num = np.array([-1, 31, 0.5]) 
  
# traversing the list 
for i in range(len(complex_num)): 
      print("{}. Number is {}".format(i + 1, complex_num[i])) 
      print ("The real part is: {}".format(complex_num[i].real)) 
      print ("The imaginary part is: {}\n".format(complex_num[i].imag)) 

Output :

1. Number is -1.0
The real part is: -1.0
The imaginary part is: 0.0

2. Number is 31.0
The real part is: 31.0
The imaginary part is: 0.0

3. Number is 0.5
The real part is: 0.5
The imaginary part is: 0.0

Attention geek! Strengthen your foundations with the Python Programming Foundation Course and learn the basics.

To begin with, your interview preparations Enhance your Data Structures concepts with the Python DS Course.

My Personal Notes

Related Articles