python中np.array的用法
时间: 2024-09-10 07:14:12 浏览: 16
在Python中,`numpy`库中的`array`函数是用来创建多维数组的主要工具,它非常类似于数学中的矩阵或向量。以下是`np.array`的基本用法:
1. 创建一维数组:
```python
import numpy as np
a = np.array([1, 2, 3, 4, 5]) # 创建包含整数的一维数组
```
2. 创建二维数组(如矩阵):
```python
b = np.array([[1, 2], [3, 4]]) # 创建一个2x2的二维数组
```
3. 数据类型转换:
```python
c = np.array([1, 'a', 3.14], dtype=float) # 将混合数据类型的列表转换为浮点型数组
```
4. 数组操作:
- 初始化:可以指定形状和值
```python
d = np.zeros((3, 3)) # 创建一个全零的3x3矩阵
e = np.ones((2, 2), dtype=int) # 创建一个全1的2x2矩阵,元素为int类型
```
- 索引、切片和广播:
```python
f = np.array([[0, 1, 2], [3, 4, 5]])
print(f[0, 1]) # 输出:1
print(f[:, ::2]) # 每隔一个元素取值,得到[0, 2, 4]
```
5. 动态维度:
```python
g = np.array([[1, 2], [3, 4, 5]]) # 第二维长度不是固定的
```