Numpy的基础用法

1.用Numpy创建数组

numpy.array(object)：创建数组，与array.array(typecode[, initializer])不同，array.array()只能创建一维数组

numpy.arange(start, stop, step, dtype=None)：创建一个从start开始，stop结束(不包含stop)，以step为步长的一维数组(step最好为整数)，dtype默认为整数(int32)

numpy.linspace(start, stop, num=50, endpoint=True, retstep=False, dtype=None, axis=0)：创建一个从start开始，stop结束（默认包含stop），默认50个元素的数组；num规定了数组元素的个数；当endpoint为False时，数组不包含stop；输出的数组元素默认为float

上述函数创建的数组全部返回ndarray类型

import numpy as np

ndarray_1 = np.array([[1,2,3], [4,5,6], (7,8,9)])
print(ndarray_1)

ndarray_2 = np.arange(1, 9)
print(ndarray_2)

ndarray_3 = np.linspace(1, 9, 9)
print(ndarray_3)

 

2.ndarray对象的方法

1.ndarray.ndim：返回数组的维数

2.ndarray.shape：返回数组的大小

3.ndarray.reshape(shape)：返回一个以shape为格式的新数组

4.ndarray.size：返回数组元素的总个数

5.ndarray.dtype：返回数组元素的类型

6.ndarray.itemsize：返回数组元素的大小（以字节为单位）

import numpy as np

x = np.linspace(1,18,18,dtype=int).reshape(2,3,3)
print(x.ndim)# 3
print(x.shape)#(2,3,3)
print(x.size)# 18
print(x.dtype)# int32
print(x.itemsize)# 4

 

3.numpy创建特殊数组(数组的元素类型默认为float)

numpy.empty(shape, dtype=float, order=‘C’)：创建一个随机数组

numpy.empty_like(prototype，dtype = None，order ='K'，subok = True，shape = None)：返回一个形状和类型与给定数组相同的新数组

shape：控制数组的形状，为int或int元组，例如：shape是3，则创建一个3x3的数组，shape是(2,3)，则创建一个2x3的数组

 order：控制存储时以行为主还是以列为主

import numpy as np

arr = np.empty((2,3))
print(arr)

a = np.arange(6).reshape(2,3)
arr_like = np.empty_like(a)
print(arr_like)

 

numpy.zeros(shape, dtype=float, order='C')：创建一个全零数组

numpy.zeros_like(a, dtype=None, order='K', subok=True, shape=None)：创建一个和给定的数组a形状和类型相同的全零新数组

import numpy as np

arr_zero = np.zeros((2,3), dtype=int)
print(arr_zero)
 
a = np.arange(6, dtype=float).reshape(2,3)
arr_zero_like = np.zeros_like(a)
print(arr_zero_like)

 

numpy.ones(shape, dtype=None, order='C')：创建一个全1数组

numpy.ones_like(a, dtype=None, order='K', subok=True, shape=None)：创建一个和给定的数组a形状和类型相同的全1数组

import numpy as np

arr_ones = np.ones((3,3), dtype=int)
print(arr_ones)

a = np.arange(9, dtype=float).reshape(3,3)
arr_ones_like = np.ones_like(a)
print(arr_ones_like)

 

numpy.full(shape, fill_value, dtype=None, order='C')：创建一个用fill_value填充的填充数组

numpy.full_like(a, fill_value, dtype=None, order='K', subok=True, shape=None)：创建一个和给定数组a形状和类型相同的填充数组

import numpy as np

arr_full = np.full((2,3), 3)
print(arr_full)

a = np.arange(6)
arr_full_like = np.full_like(a, 1)
print(arr_full_like)

 

numpy.eye(N, M=None, k=0, dtype=, order='C')：创建一个对角线元素为1，其余元素为0的数组

M：默认等于N，就是创建一个NxN的数组

k：k为正数，则对角线上移k位；k为负数，则对角线下移k位；若|k|>=M，则数组元素全为0

numpy.identity(n, dtype=None)：创建一个nxn的，对角线元素为1的数组，元素类型默认为float

import numpy as np

arr_eye = np.eye(3)
print(arr_eye)

arr_eye2 = np.eye(3, k=1)
print(arr_eye2)

arr_eye3 = np.eye(3, k=3)
print(arr_eye3)

arr_id = np.identity(3)
print(arr_id)

 

4.数组的索引

4.1 一维数组的索引

一维数组的索引与列表的索引类似

import numpy as np

a = np.arange(10)
print(a[-1])# 9
print(a[5:])#[5 6 7 8 9]
print(a[-5:10])#[5 6 7 8 9]
# 第一个参数小于第二个时，最后一个参数要为负数，表示从后向前索引
print(a[-2:2:-1])#[8 7 6 5 4 3]
# 翻转
print(a[::-1])#[9 8 7 6 5 4 3 2 1 0]
print(a[5::2])#[5 7 9]
# 整数索引,索引数组中的第2个，倒数第4个，第5个元素
print(a[[2,-4,5]])#[2 6 5]

# 下标索引方式和原数组占用同样的存储空间，修改b时，a也会被修改
b = a[1:3]
b[0] = 100
print(a)#[0,100,2,3,4,5,6,7,8,9]
# 列表索引和原数组不占用同样的存储空间，修改c时，a2不会被修改
a2 = np.arange(10)
c = a2[[0,1,2]]
c[0] = 100
print(a2)#[0,1,2,3,4,5,6,7,8,9]

 

4.2 二维数组的索引

import numpy as np

a = np.arange(24).reshape(4,6)
# 取第一、二行的第一个元素
print(a[0,0], a[1,0]) 
# 取第一行的所有元素，第一行和第一列的索引下标为0，以此类推 
print(a[0])
# 取第一列的所有元素
print(a[:, 0])
# 取第一行到第三行的部分元素
print(a[0:3, :3])
# 翻转
print(a[::-1, ::-1])

#a[i:j, i:j] 逗号前面的表示取行，逗号后面的表示取列

 

4.3 三维数组的索引

import numpy as np

a = np.arange(24).reshape(2,3,4)
# 可以把数组a看做一个2层楼的房子，每层有3x4个房间，第一个索引值表示位于哪一个楼层，后面的两个索引与二维数组类似
print(a[0, :, :])
# 当后面都为:时，可以用一个...代替
print(a[0,...])
# 索引第一层的第一行，第一列的元素
print(a[0, 0, 0])
# 索引两层的最后一列元素
print(a[:,:,-1])

 

5.Numpy.random模块

numpy.random.rand(d0, d1, d2,..., dn)：生成一个元素值为0-1之间的随机数组，数组的形状由传入的参数决定：当不传入参数时，生成一个元素；传入一个参数时，根据参数的大小生成一个一维数组；传入两个参数时，生成一个二维数组

numpy.random.randn(d0,d1,d2,...,dn)：生成一个服从标准正态分布的数或数组，数组的形状由传入的参数(d1,d2,...dn)决定

import numpy as np

arr1 = np.random.rand()
print(arr1, '\n')
arr2 = np.random.rand(6)
print(arr2, '\n')
arr3 = np.random.rand(2,3)
print(arr3, '\n')
arr4 = np.random.randn(2,3,4)
print(arr4)

 

numpy.random.randint(low, high=None, size=None, dtype='l')：从小到大返回随机整数，如果high=None，则整数的范围是[0, low)，否则为[low, high)；若size=None，则生成一个随机整数，否则根据size来生成相对应的数组，下同 

import numpy as np

arr1 = np.random.randint(5)
print(arr1)
arr2 = np.random.randint(1, 10, (2,3))
print(arr2)
# 当size前面只有一个参数时，size参数要写成size=(a,b)
arr3 = np.random.randint(10,size=(2,3))
print(arr3)

 

numpy.random.normal(loc=0.0, scale=1.0, size=None)：生成一个服从正态分布的数或数组，loc为正态分布的均值，scale为正态分布的标准差

numpy.random.uniform(low=0.0, high=1.0, size=None)：生成一个服从均匀分布的数或数组，生成的所有值都大于等于low，小于high

numpy.random.poisson(lam=1.0, size=None)：生成一个服从lam=1.0的泊松分布的数或数组

import numpy as np

arr1 = np.random.normal(0, 1, (2,3))# 服从标准正态分布
print(arr1)
arr2 = np.random.uniform(3, 5, (2,3))
print(arr2)
arr3 = np.random.poisson(size=(2,3))
print(arr3)

 

numpy.random.choice(a, size=None, replace=True, p=None)：从给定的一维数组a中生成一个随机样本，replace设定样本是否更换；p设定a中每个数被选中的概率，未给出则每个数服从均匀分布；

import numpy as np

a = np.arange(5)
arr1 = np.random.choice(a, (2,3))
print(arr1)
arr2 = np.random.choice(a, 5, p=[0.1,0.2,0,0.7,0])
print(arr2)

 

numpy.random.shuffle(a)：随机打乱一个数组，多维数组则仅沿第一个轴随机排列，函数返回None，原数组被改变 

import numpy as np

a = np.arange(12)
np.random.shuffle(a)
print(a)

b = np.arange(12).reshape(3,4)
np.random.shuffle(b)
print(b)

 

 numpy.random.seed(seed=None)：用于设定随机数种子

 

6.统计函数

Numpy可以用于求数组的和与平均值

numpy.sum(a, axis=None, dtype=None, out=None, keepdims=, initial=): 用于求数组元素的和，axis=0：求每列的和，axis=1：求每行的和

import numpy as np

data1 = np.sum(np.arange(10).reshape(2,5))
print(data1)

data2 = np.sum(np.arange(10).reshape(2,5), axis=0)
print(data2)

data3 = np.sum(np.arange(10).reshape(2,5), axis=1)
print(data3)

data = np.sum([0.5, 0.7, 0.2, 1.5], dtype=np.int32)#手册的例子，不懂结果为什么为1
print(data)

 

numpy.mean(a, axis=None, dtype=None, out=None, keepdims=)：用于求均值

import numpy as np

data1 = np.mean(np.arange(10).reshape(2,5))
print(data1)

data2 = np.mean(np.arange(10).reshape(2,5), axis=0)
print(data2)

data3 = np.mean(np.arange(10).reshape(2,5), axis=1)
print(data3)

 

numpy.average(a, axis=None, weights=None, returned=Flase)：用于求加权平均值，weights是用于加权的数组

import numpy as np

data = np.average([5,10],weights=[1,2])#相当于(1*5+2*10)/(1+2)
print(data)

 numpy.var(a，axis=None)：用于求方差；

numpy.std(a, axis=None)：用于求标准差;

numpy.max(a, axis=None)：用于求最大值；

numpy.min(a，axis=None)：用于求最小值；

 

7.矩阵运算（用matlab更好一点）

Numpy用于矩阵运算的模块是numpy.linalg

numpy.transpose(a, axes=None)：用于矩阵转置，或者可以直接用 ndarray.T

numpy.dot(a, b)：一维时计算向量内积；二维时计算矩阵乘法，或者 ndarray.dot(b)

numpy.trace(a，offset=0)：用于计算矩阵的对角线元素之和

linalg.det(a)：用于计算行列式

linalg.eig(a)：用于计算方阵的特征值和特征向量

linalg.inv(a)：用于求逆矩阵

linalg.solve(a, b)：解线性方程组Ax=b

linalg.qr(a,mode='reduced')：用于矩阵的QR分解

linalg.svd(a,full_matrices=1,compute_uv=1)：用于矩阵的奇异值分解

 

 更多用法请参考：Numpy手册