如何生成复数型numpy数组
时间: 2023-10-13 15:43:18 浏览: 1112
python numpy数组中的复制知识解析
要生成复数型的NumPy数组,你可以使用`numpy()`函数并传入包含复数的列表或使用`numpy.zeros()`函数创建一个指定形状的复数型数组。
下面是两种方法的示例:
1. 使用`numpy.array()`函数:
```python
import numpy as np
# 生成包含复数的列表
data = [1 + 2j, 3 - 4j, 5 + 6j]
# 将列表转换为复数型NumPy数组
arr = np.array(data, dtype=complex)
print(arr)
```
输出:
```
[1.+2.j 3.-4.j 5.+6.j]
```
2. 使用`numpy.zeros()`函数:
```python
import numpy as np
# 创建一个指定形状的复数型数组
shape = (3, 2) # 3行2列
arr = np.zeros(shape, dtype=complex)
print(arr)
```
输出:
```
[[0.+0.j 0.+0.j]
[0.+0.j 0.+0.j]
[0.+0.j 0.+0.j]]
```
这两种方法都可以用来生成复数型的NumPy数组。你可以根据需要选择适合的方法。
阅读全文