Skip to content
Related Articles

Related Articles

Extracting the real and imaginary parts of an NumPy array of complex numbers
  • Last Updated : 02 Sep, 2020
GeeksforGeeks - Summer Carnival Banner

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 arrow_drop_up
Recommended Articles
Page :