Numpy的常用语法

#数组创建

a = np.array([[1,2],[3,4],[5,6]])#创建3行2列二维数组。
print(a)
array([[1, 2],
       [3, 4],
       [5, 6]])
a = np.zeros(6)#创建长度为6的，元素都是0一维数组
a = np.zeros((2,3))#创建3行2列，元素都是0的二维数组
a = np.ones((2,3))#创建3行2列，元素都是1的二维数组
a = np.empty((2,3)) #创建3行2列,未初始化的二维数组
a = np.arange(6)#创建长度为6的，元素都是0一维数组array([0, 1, 2, 3, 4, 5])
a = np.arange(1,7,1)#结果与np.arange(6)一样。第一，二个参数意思是数值从1〜6,不包括7.第三个参数表步长为1.
a=np.linspace(0,10,7) # 生成首位是0，末位是10，含7个数的等差数列[  0.           1.66666667   3.33333333   5.         6.66666667  8.33333333  10.        ]
a=np.logspace(0,4,5)#用于生成首位是10**0，末位是10**4，含5个数的等比数列。[  1.00000000e+00   1.00000000e+01   1.00000000e+02   1.00000000e+03 1.00000000e+04]

#数组合并

a = np.array([[1,2],[3,4],[5,6]])
b = np.array([[10,20],[30,40],[50,60]])
np.vstack((a,b))
array([[ 1,  2],
       [ 3,  4],
       [ 5,  6],
       [10, 20],
       [30, 40],
       [50, 60]])
np.hstack((a,b))
array([[ 1,  2, 10, 20],
       [ 3,  4, 30, 40],
       [ 5,  6, 50, 60]])

#不同维数相加(a,b)+(b,c)广播=(a,c)

a = np.array([[1],[2]])
print(a)
array([[1],
       [2]])
	   
b=([[10,20,30]])#生成一个list，注意，不是np.array。
print(b)
[[10, 20, 30]]

print(a+b)
array([[11, 21, 31],
       [12, 22, 32]])
	   
c = np.array([10,20,30])
print(c)
array([10, 20, 30])

print(c.shape)
(3,)

print(a+c)
array([[11, 21, 31],
       [12, 22, 32]])

#删除行列

a = np.array([[1,2],[3,4],[5,6]])
np.delete(a,1,axis = 0)#删除a的第二行。
array([[1, 2],
       [5, 6]])
	   
np.delete(a,(1,2),0)#删除a的第二，三行。
array([[1, 2]])

np.delete(a,1,axis = 1)#删除a的第二列。
array([[1],
       [3],
       [5]])

#argmax 返回的是最大数的索引

import numpy as np
a = np.array([3, 1, 2, 4, 6, 1])
print(np.argmax(a)) #4

import numpy as np
a = np.array([[1, 5, 5, 2],
              [9, 6, 2, 8],
              [3, 7, 9, 1]])
print(np.argmax(a, axis=0))  #(1,2,2,1)
print(np.argmax(a, axis=1))  #(1,0,2)

#array 进行强制类型转换

a.astype(float)