numpy的一些基础用法

import numpy as np

x = np.array([[[1]]])

x.ndim维度，x.shape形状

print(x,x.ndim,x.shape)

创建数组

x = np.zeros((2,3,4),dtype=np.int8)
print(x)

创建1矩阵

x = np.ones((2,3),dtype=np.int8)
print(x)

使用arange生成连续的元素

x = np.arange(6)
print(x)
x = np.arange(1,6)
#复制数组并转换类型，astype（讲字符串转数值）
x = np.array([1,2,3],dtype=np.float32)
y = x.astype(dtype=np.int8)
print(x,y)
#使用其他的数组的数据类型作为参数
x = np.array([1,2,3],dtype=np.float64)
y = np.arange(3,dtype=np.int32)
print("->",y)
y = y.astype(x.dtype)
print("=>",y)

ndarray数组与标量/数组的运算

x = np.array([1,2,3])
print(x2)
print(x>2)
y = np.array([2,3,4])
print(xy)
print(x>y)
#ndarray的基本索引
x = np.array([[1,2],[3,4],[5,6]])
print(x.shape)
print(x[1][0],x[1,0])
x = np.array([[[1,2],[3,4]],[[5,6],[7,8]]])
print(x.shape)
print(x[0,1,0])
y = x[0].copy()
print(y)

ndarray的切片（切片中，逗号隔开表示进入下一个维度）

一维

x = np.array([1,2,3,4,5,6])
print(x[1:4])
print(x[0:4:2])#步长为2
#二维
x = np.array([[[1,2],[3,4]],[[5,6],[7,8]]])
print(x)
print(“1”,x[:2])
print(“2”,x[:2,:1,:1])
print(“3”,x[:2][:1][:1])
print(“4”,x[0,0,1],x[0][0][1])

ndarray的布尔类型索引

x = np.array([3,2,1])
y = np.array([True,True,False])
print(x[y])
print(x[y == False])
print(x>=3)
print(x[x>=3])
print(x[(x2)|(x1)])

花式索引：使用整型数组作为索引

x = np.array([1,2,3,4])
print(x[[0,1,2]])
print(x[[-1,-2,-3]])
print("---------------------")
x = np.array([[1,2],[3,4],[5,6]])
print(x[[0,1]])
print("---------------------")

print(x[[[0,1],[0,1]]])

print(x[[[0,1],[0]]])

print(x[[0,1]][:1,[0,1]])
x = np.arange(24).reshape(2,3,4)
print(x)
print("---------------------")

数组的转置和轴对象

k = np.arange(12).reshape((3,4))
print(k)
print(k.T)

内积

k = np.dot(k,k.T) #相当于长乘宽
print(“内积：”,k)