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

  • Post category:Python

提取NumPy复数数组的实部和虚部可以通过以下两个方法实现:

  1. 通过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.]
  1. 通过数组切片的方式分别提取实部和虚部,使用方法如下:
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函数来分别提取实部和虚部,输出的结果分别为实数和虚数。在第二个示例中,使用了数组切片的方式来提取实部和虚部,输出的结果同样分别为实数和虚数。