《Python数据分析实战》3 NumPy库
- 初始化数组
arr = np.array([[1,2],[3,4]])
arr = np.array([[1,2],[3,4]], dtype=complex)
arr = np.zeros((3,4))
arr = np.ones((1,2))
arr = np.arange(4,10).reshape(2,3)
arr = np.linspace(0,1,6) #[0.0,0.2,0.4,0.6,0.8,1.0]
arr = np.random.random(3)
arr = np.random.random((5,2))
- 数组内置函数
type(arr):numpy.ndarray
arr.dtype:int32、float64……
arr.ndim:2
arr.size:4
arr.shape:(2L,2L)
arr.itemsize:4
- 数组计算函数
arr+1:元素和
arr*2:元素积
arr1+arr2:元素和
arr1*arr2:元素积
np.dot(arr1, arr2):矩阵积
sin(arr):元素三角函数
sqrt(arr):元素开方
arr.sum():所有元素和
arr.min():所有元素最小值
arr.max():所有元素最大值
arr.mean():所有元素平均值
- 数组迭代函数
np.apply_along_axis(np.mean, axis=0, arr=arr):按列迭代
np.apply_along_axis(lambda x:x*2, axis=1, arr=arr):按行迭代
- 布尔数组:
arr<0.5
- 拼接数组:
np.vstack((arr1, arr2)):垂直入栈
np.hstack((arr1, arr2)):水平入栈
- 切分数组:
np.vsplit(arr, n):垂直平分n部分
np.hsplit(arr, n):水平平分n部分
np.split(arr, [1,3,5], axis=1):垂直分割
- 数组保存/读取
np.save('data.npy', arr)
arr = np.load('data.npy')