python中复数矩阵怎么分存储为实部和虚部
时间: 2024-05-04 07:18:14 浏览: 18
在Python中,复数矩阵可以使用NumPy库中的数组来表示。如果要将复数矩阵分别存储为实部和虚部,可以使用NumPy库中的real和imag函数来获取实部和虚部矩阵。
以下是一个示例代码:
```python
import numpy as np
# 创建一个3x3的复数矩阵
complex_matrix = np.array([[1+2j, 2+3j, 3+4j],
[4+5j, 5+6j, 6+7j],
[7+8j, 8+9j, 9+10j]])
# 获取实部矩阵
real_matrix = np.real(complex_matrix)
# 获取虚部矩阵
imag_matrix = np.imag(complex_matrix)
print("复数矩阵:\n", complex_matrix)
print("实部矩阵:\n", real_matrix)
print("虚部矩阵:\n", imag_matrix)
```
输出结果如下所示:
```
复数矩阵:
[[ 1. +2.j 2. +3.j 3. +4.j]
[ 4. +5.j 5. +6.j 6. +7.j]
[ 7. +8.j 8. +9.j 9.+10.j]]
实部矩阵:
[[ 1. 2. 3.]
[ 4. 5. 6.]
[ 7. 8. 9.]]
虚部矩阵:
[[ 2. 3. 4.]
[ 5. 6. 7.]
[ 8. 9. 10.]]
```