提取NumPy复数数组的实部和虚部可以通过以下两个方法实现:
- 通过numpy.real和numpy.imag函数分别提取实部和虚部,使用方法如下:
import numpy as np
# 创建一个复数数组
a = np.array([1+2j, 3+4j, 5+6j])
# 使用numpy.real函数提取实部
real_part = np.real(a)
print("实部:", real_part)
# 使用numpy.imag函数提取虚部
imag_part = np.imag(a)
print("虚部:", imag_part)
输出结果:
实部: [1. 3. 5.]
虚部: [2. 4. 6.]
- 通过数组切片的方式分别提取实部和虚部,使用方法如下:
import numpy as np
# 创建一个复数数组
a = np.array([1+2j, 3+4j, 5+6j])
# 提取实部和虚部
real_part = a.real
print("实部:", real_part)
imag_part = a.imag
print("虚部:", imag_part)
输出结果:
实部: [1. 3. 5.]
虚部: [2. 4. 6.]
以上两种方法都可以用来提取NumPy复数数组的实部和虚部。
示例1:
import numpy as np
# 创建一个复数数组
a = np.array([1.5-2.3j, 3.2+4.7j, 5.6-6.1j])
# 使用numpy.real函数提取实部
real_part = np.real(a)
print("实部:", real_part)
# 使用numpy.imag函数提取虚部
imag_part = np.imag(a)
print("虚部:", imag_part)
输出结果:
实部: [ 1.5 3.2 5.6]
虚部: [-2.3 4.7 -6.1]
示例2:
import numpy as np
# 创建一个复数数组
a = np.array([2j, 5j, 8j])
# 提取实部和虚部
real_part = a.real
print("实部:", real_part)
imag_part = a.imag
print("虚部:", imag_part)
输出结果:
实部: [0. 0. 0.]
虚部: [2. 5. 8.]
以上两个示例展示了提取NumPy复数数组实部和虚部的方式。在第一个示例中,使用了numpy.real和numpy.imag函数来分别提取实部和虚部,输出的结果分别为实数和虚数。在第二个示例中,使用了数组切片的方式来提取实部和虚部,输出的结果同样分别为实数和虚数。