numpy 等距数组
时间: 2023-11-06 19:06:25 浏览: 81
您可以使用NumPy库中的`arange`函数来创建等距数组。`arange`函数接受起始值、终止值和步长作为参数,并返回一个包含对应范围内等距元素的一维数组。
以下是一个使用NumPy创建等距数组的示例代码:
```python
import numpy as np
start = 0
stop = 10
step = 2
arr = np.arange(start, stop, step)
print(arr)
```
在上述示例中,我们设置起始值为0,终止值为10,步长为2。运行结果将是一个包含0、2、4、6、8的一维数组。
如果您想创建一个包含等距元素的二维数组,可以使用`meshgrid`函数。`meshgrid`函数接受两个一维数组作为输入,并返回两个二维数组,其中一个数组的行数是第一个输入数组的长度,列数是第二个输入数组的长度,另一个数组则相反。
以下是一个使用NumPy创建等距二维数组的示例代码:
```python
import numpy as np
x = np.arange(3) # 一维数组 [0, 1, 2]
y = np.arange(4) # 一维数组 [0, 1, 2, 3]
X, Y = np.meshgrid(x, y)
print(X)
print(Y)
```
在上述示例中,我们创建了两个一维数组x和y,分别包含0到2和0到3的元素。`meshgrid`函数将这两个一维数组转换为对应的二维数组X和Y。运行结果将分别是:
```
[[0 1 2]
[0 1 2]
[0 1 2]
[0 1 2]]
[[0 0 0]
[1 1 1]
[2 2 2]
[3 3 3]]
```
这里的X数组表示在x轴上重复4次,Y数组表示在y轴上重复3次,从而形成一个包含12个等距点的二维数组。
阅读全文