np.array的用法
时间: 2024-06-22 19:03:02 浏览: 171
python将txt文件读入为np.array的方法
`numpy.array`是NumPy库中的核心数据结构,它是一个强大的多维数组,用于处理数值计算。使用`np.array`,你可以创建和操作一系列相同类型的元素,并且它提供了许多高效的数学运算功能。以下是一些基本的用法:
1. 创建数组:你可以直接传递列表、元组或类似的数据,`np.array()`会根据这些数据自动推断数据类型。
```python
import numpy as np
# 从列表创建数组
arr = np.array([1, 2, 3, 4, 5])
```
2. 数组形状:`array`的形状可以通过`shape`属性获取或改变。
```python
arr.shape # 输出: (5,)
arr.reshape(2, 2) # 改变形状为2x2
```
3. 数据类型:`dtype`属性显示数组的元素类型。
```python
arr.dtype # 输出: int64 或者 dtype('int64')
```
4. 数组操作:NumPy提供大量的数学函数和方法,如加法、乘法、索引、切片等。
```python
arr + 2 # 增加每个元素2
arr * arr # 数组的元素自乘
arr[0:3] # 获取前三个元素
```
5. 初始值和维度:可以指定初始值和维度来创建数组。
```python
arr = np.zeros((3, 3)) # 创建一个3x3全零矩阵
arr = np.ones((2, 2), dtype='float32') # 创建一个2x2全1的浮点型矩阵
```
阅读全文