numpy的用法
import numpy as np ndarray #创建数组,array data1=[6,7,8,0,1] np.array(data1) #一维数组 array([6, 7, 8, 0, 1]) #创建多维数组 data2=[[1,2,3,4],[5,6,7,8]] arr1=np.array(data2) print(data2) [[1, 2, 3, 4], [5, 6, 7, 8]] #dtype数据类型 arr1.dtype dtype('int32') #shape属性 arr1.shape (2, 4) arr2=np.array(data2,dtype=np.int64) print(arr2) print(arr2.dtype) [[1 2 3 4] [5 6 7 8]] int64 #float data3=[6,7,5,8,0,1] arr3=np.array(data3) print(arr3) print(arr3.dtype) [6 7 5 8 0 1] int32 #astype类型转换 arr3.astype(np.int64) array([6, 7, 5, 8, 0, 1], dtype=int64) #ones np.ones(10) array([1., 1., 1., 1., 1., 1., 1., 1., 1., 1.])