Numpy常用函数

1. 简介

2. numpy.meshgrid

• 功能：meshgrid函数用两个坐标轴上的点在平面上画网格
• 定义

　　[X,Y]=meshgrid(x,y)
　　[X,Y]=meshgrid(x)       #与[X,Y]=meshgrid(x,x)是等同的
　　[X,Y,Z]=meshgrid(x,y,z) #生成三维数组，可用来计算三变量的函数和绘制三维立体图

• meshgrid(x,y)
  • 将向量x和y定义的区域转换成矩阵X和Y,其中矩阵X的行向量是向量x的简单复制，而矩阵Y的列向量是向量y的简单复制
  • 假设x是长度为m的向量，y是长度为n的向量，则最终生成的矩阵X和Y的维度都是 nm （注意不是mn）
• 实例

import numpy as np
import matplotlib.pyplot as plt
%matplotlib inline
m, n = (5, 3)
x = np.linspace(0, 1, m)
y = np.linspace(0, 1, n)
X, Y = np.meshgrid(x,y)
# x=[ 0.  ,  0.25,  0.5 ,  0.75,  1.  ]
# y=[ 0. ,  0.5,  1. ]
# X = [[ 0.  ,  0.25,  0.5 ,  0.75,  1.  ],
#      [ 0.  ,  0.25,  0.5 ,  0.75,  1.  ], 
#      [ 0.  ,  0.25,  0.5 ,  0.75,  1.  ]]  
# Y= [[ 0. ,  0. ,  0. ,  0. ,  0. ],
#     [ 0.5,  0.5,  0.5,  0.5,  0.5],
#     [ 1. ,  1. ,  1. ,  1. ,  1. ]]

3. numpy.vstack

• 功能：沿着竖直方向将矩阵堆叠起来
• 示例

import numpy as np

arr1 = np.array([1, 2, 3])
arr2 = np.array([4, 5, 6])
res = np.vstack((arr1, arr2))
#array([[1, 2, 3],
#       [4, 5, 6]])

4. numpy.hstack

• 功能：沿着水平方向将数组堆叠起来
• 示例

import numpy as np
arr1 = np.array([1, 2, 3])
arr2 = np.array([4, 5, 6])
res = np.hstack((arr1, arr2))
print res
# [1 2 3 4 5 6]
arr1 = np.array([[1, 2], [3, 4], [5, 6]])
arr2 = np.array([[7, 8], [9, 0], [0, 1]])
res = np.hstack((arr1, arr2))
print res
#[[1 2 7 8]
# [3 4 9 0]
# [5 6 0 1]]