NumPy笔记

import numpy as np

1、创建

data1 = np.array([1,2,3,4]) -- array([1,2,3,4])

data2 = np.arange(4) -- array([0,1,2,3])

data3 = np.arange(9).reshape((3,3))

2、维度

data1.shape -- 各维度

data1.dtype -- 数据类型

3、索引和切片

data1[0] -- 1

data1[1:3] -- array([2,3])

data3[0] -- array([0,1,2])

data3[0][0] -- 0

data3[:2,1:] -- array([[1,2],[4,5]])

data3[[0,1],[1,2]] -- array([[1,2],[4,5]])

4、条件判断

np.where(data3>4,4,data3) -- array([[0,1,2],[3,4,4],[4,4,4]])

5、统计计算

arr = np.random.randn(5,4)

arr.mean() -- 均值（等价于np.mean(arr)）

arr.sum() -- 求和

arr.min()

arr.max()

6、排序

arr.sort()

7、唯一化

arr1 = np.array([1,2,3,1,2])

np.unique(arr1) -- array([1,2,3])

8、随机数

np.random(10)

np.random.normal(size(2,2)) -- 标准正态分布样本