使用arange、ones、eye等函数创建一维或多维数组;
时间: 2024-10-19 13:13:21 浏览: 28
python中numpy的矩阵、多维数组的用法
在Python的NumPy库中,有几种常用的函数用于生成一维或多维数组:
1. **arange()** 函数:它会按照指定的步长生成一系列等差数列。例如,`numpy.arange(start, stop, step)` 会返回从 `start` (包含) 到 `stop` (不包含) 的整数序列,默认步长为1。如果需要浮点数序列,可以将 `step` 设置为一个小数值。
2. **ones()** 函数:这个函数用于创建所有元素都为1的数组,可以指定形状。如 `numpy.ones(shape)`,其中 `shape` 可以为单个数字(生成一维数组),也可以是一个元组(生成多维度数组)。
3. **eye()** 函数:用于生成单位矩阵,一般用于表示对角线元素为1,其他元素为0的方阵。如 `numpy.eye(n)` 会生成一个n阶的单位矩阵,如果需要生成m×n矩阵,可以给定两个参数,如 `numpy.eye(m, n)`。
举例来说,如果你想要创建一个长度为5的一维数组,每个元素都是1,你可以这样做:
```python
import numpy as np
one_dim_array = np.ones(5)
```
或者创建一个3x3的单位矩阵:
```python
unit_matrix = np.eye(3)
```
阅读全文