要提取NumPy复数数组的实部和虚部,可以使用real
和imag
属性。下面是详细的攻略:
1. 创建NumPy复数数组
首先,我们需要创建一个包含复数数值的NumPy数组。可以使用numpy.array
函数,也可以使用随机数生成函数等方式创建。
import numpy as np
# 创建复数数组
arr = np.array([1+2j, 3+4j, 5+6j])
2. 提取实部和虚部
使用real
和imag
属性可以分别提取数组的实部和虚部。
# 提取实部和虚部
real = arr.real
imag = arr.imag
print("实部:", real) # [1. 3. 5.]
print("虚部:", imag) # [2. 4. 6.]
以上代码分别将实部和虚部保存到real
和imag
变量中,并使用print
函数输出结果。可以看到,real
和imag
变量的结果分别为实部[1. 3. 5.]和虚部[2. 4. 6.]。
示例1:直接提取一个复数的实部和虚部
x = 6 + 4j
real_part = x.real
imag_part = x.imag
print("实部:", real_part) # 6.0
print("虚部:", imag_part) # 4.0
以上代码提取了复数6+4j的实部6.0和虚部4.0。
示例2:提取一个包含复数数值的NumPy数组的实部和虚部
import numpy as np
# 创建复数数组
arr = np.array([1+2j, 3+4j, 5+6j])
# 提取实部和虚部
real = arr.real
imag = arr.imag
print("实部:", real) # [1. 3. 5.]
print("虚部:", imag) # [2. 4. 6.]
以上代码创建了一个包含复数数值的NumPy数组,并提取了该数组的实部和虚部。此处也使用print
函数输出结果。可以看到,实部和虚部的提取成功。
本站文章如无特殊说明,均为本站原创,如若转载,请注明出处:提取NumPy复数数组的实部和虚部 - Python技术站