提取 NumPy 复数数组的实部和虚部

Numpy 库为我们提供了诸如real()和imag()之类的函数来查找复数的实部和虚部。

• real() :求复数的实部
• imag() :求复数的虚部

示例 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))

输出 ：

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

例 2：实数的虚数为 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))

输出 ：

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

在评论中写代码？请使用 ide.geeksforgeeks.org，生成链接并在此处分享链接。