一/安装
pip install numpy
二/引用
import numpy as np
三/使用
1.数组
- 创建一个ndarray数组
data1=np.array([1,2,3,4,5])
data2=np.array([[1,2,3],[4,5,6]])
- 数组维度
print(data2.ndim)
- 数组维度长度(几行几列)
print(data2.shape)
- 创建一个全是0的数组
data=np.zeros(10)
- 创建一个全是1的二维数组
data=np.ones((2,3))
- 获取数组中某个数字,用索引
data=np.arrange(10)
print(data[5])
data2=np.array([[1,2,3],[4,5,6]])
print(data2[0][1])
print(data2[0,1])
- 获取数组中某几个数字，用切片
data=np.arrange(10)
print(data[3:6])
注意：切换后得到的数据还是原始数据，任何修改都反应到原始数据上
dataslice[2]=99
如果需要不影响原始数据，切片时使用
data[3:6].copy()
- 变换数组的维度
data=np.arrange(10)
print(data)
从一维变成二维
print(data.reshape((2,5)))
转置
print(data.reshape((2,5)).T)
- 对数组元素求平方根
data=np.arrange(10)
print(np.sqrt(data))
其他一些方法
| 函数 | 说明 |
| abs | 求绝对值 |
| square | 求平方 |
| ceil | 向上取整，计算大于等于该元素的最小整数 |
| floor | 向下取整，计算小于等于该元素的最大整数 |
| isnan | 计算哪些元素是非数字 |
| sign | 计算正负号，1 0 -1 |
2.数组运算
- 两个数组相加
data1=np.array([1,2,3])
data2=np.array([4,5,6])
print(data1+data2)
print(np.add(data1,data2))
其他方法：substract，multiply，divide，power，fmax，fmin
- 求和
data=np.arrange(10)
print(data.sum())
- 求平均值
print(data.mean())
其他统计：std，min,max,argmin,argmax
- 数组排序
data.sort()
3.其他
- 文件读取
data=np.genfromtxt('data.txt',delimiter=',')
- 类型转换
print(data.astype(int))
