1 # -*- coding: utf-8 -*- 2 3 ############################################################################### 4 ####################### 正文代码 ####################### 5 ############################################################################### 6 7 # 代码 2-1 8 import numpy as np #导入 NumPy 库 9 arr1 = np.array([1, 2, 3, 4]) #创建一维数组 10 print(' 创建的数组为: ',arr1) 11 12 # 创建二维数组 13 arr2 = np.array([[1, 2, 3, 4],[4, 5, 6, 7], [7, 8, 9, 10]]) 14 print('创建的数组为:\n',arr2) 15 16 print('数组类型为:',arr2.dtype) #查看数组类型 17 print('数组元素个数为:',arr2.size) #查看数组元素个数 18 print('数组每个元素大小为:',arr2.itemsize) #查看数组每个元素大小 19 20 21 # 代码 2-2 22 arr2.shape = 4,3 #重新设置shape 23 print('重新设置shape 后的arr2 为:',arr2) 24 25 # 代码 2-3 26 print('使用arange函数创建的数组为:\n',np.arange(0,1,0.1)) 27 28 # 代码 2-4 29 print('使用linspace函数创建的数组为:',np.linspace(0, 1, 12)) 30 31 # 代码 2-5 32 print('使用logspace函数创建的数组为:',np.logspace(0, 2, 20)) 33 34 # 代码 2-6 35 print('使用zeros函数创建的数组为:',np.zeros((2,3))) 36 37 # 代码 2-7 38 print('使用eye函数创建的数组为:',np.eye(3)) 39 40 # 代码 2-8 41 print('使用diag函数创建的数组为:',np.diag([1,2,3,4])) 42 43 # 代码 2-9 44 print('使用ones函数的数组为:',np.ones((5,3))) 45 46 # 代码 2-10 47 print('转换结果为:',np.float64(42)) #整型转换为浮点型 48 print('转换结果为:',np.int8(42.0)) #浮点型转换为整型 49 print('转换结果为:',np.bool(42)) #整型转换为布尔型 50 print('转换结果为:',np.bool(0)) #整型转换为布尔型 51 print('转换结果为:',np.float(True)) #布尔型转换为浮点型 52 print('转换结果为:',np.float(False)) #布尔型转换为浮点型 53 54 # 代码 2-11 55 df = np.dtype([("name", np.str_, 40), ("numitems", np.int64), 56 ("price",np.float64)]) 57 print('数据类型为:',df) 58 59 # 代码 2-12 60 print('数据类型为:',df["name"]) 61 print('数据类型为:',np.dtype(df["name"])) 62 63 # 代码 2-13 64 itemz = np.array([("tomatoes", 42, 4.14),("cabbages", 13, 1.72)], 65 dtype=df) 66 print('自定义数据为:',itemz) 67 68 # 代码 2-14 69 print('生成的随机数组为:',np.random.random(100)) 70 71 # 代码 2-15 72 print('生成的随机数组为:\n',np.random.rand(10,5)) 73 74 # 代码 2-16 75 print('生成的随机数组为:\n',np.random.randn(10,5)) 76 77 # 代码 2-17 78 print('生成的随机数组为:',np.random.randint(2,10,size = [2,5])) 79 80 # 代码 2-18 81 arr = np.arange(10) 82 print('索引结果为:',arr[5]) #用整数作为下标可以获取数组中的某个元素 83 #用范围作为下标获取数组的一个切片,包括arr[3]不包括arr[5] 84 print('索引结果为:',arr[3:5]) 85 print('索引结果为:',arr[:5]) #省略开始下标,表示从arr[0]开始 86 #下标可以使用负数,-1表示从数组后往前数的第一个元素 87 print('索引结果为:',arr[-1]) 88 89 arr[2:4] = 100,101 90 print('索引结果为:',arr) #下标还可以用来修改元素的值 91 #范围中的第三个参数表示步长,2表示隔一个元素取一个元素 92 print('索引结果为:',arr[1:-1:2]) 93 print('索引结果为:',arr[5:1:-2]) #步长为负数时,开始下标必须大于结束下标 94 95 # 代码 2-19 96 arr = np.array([[1, 2, 3, 4, 5],[4, 5, 6, 7, 8], [7, 8, 9, 10, 11]]) 97 print('创建的二维数组为:',arr) 98 print('索引结果为:',arr[0,3:5]) #索引第0行中第3和第4列的元素 99 #索引第2和第3行中第3列、第4列和第5列的元素 100 print('索引结果为:\n',arr[1:,2:]) 101 print('索引结果为:',arr[:,2]) #索引第2列的元素 102 103 # 代码 2-20 104 #从两个序列的对应位置取出两个整数组成下标:arr[0,1], arr[1,2], arr[2,3] 105 print('索引结果为:',arr[[(0,1,2),(1,2,3)]]) 106 print('索引结果为:',arr[1:,(0,2,3)]) #索引第2、3行中第0、2、3列的元素 107 108 mask = np.array([1,0,1],dtype = np.bool) 109 #mask是一个布尔数组,它索引第1、3行中第2列的元素 110 print('索引结果为:',arr[mask,2]) 111 112 113 # 代码 2-21 114 ''' 115 arr = np.arange(12) #创建一维数组 116 print('创建的一维数组为:',arr) 117 print('新的一维数组为:',arr.reshape(3,4)) #设置数组的形状 118 print('数组维度为:',arr.reshape(3,4).ndim) #查看数组维度 119 ''' 120 121 # 代码 2-22 122 arr = np.arange(12).reshape(3,4) 123 print('创建的二维数组为:',arr) 124 print('数组展平后为:',arr.ravel()) 125 126 # 代码 2-23 127 print('数组展平为:',arr.flatten()) #横向展平 128 print('数组展平为:',arr.flatten('F')) #纵向展平 129 130 # 代码 2-24 131 arr1 = np.arange(12).reshape(3,4) 132 print('创建的数组1为:',arr1) 133 134 arr2 = arr1*3 135 print('创建的数组2为:',arr2) 136 print('横向组合为:',np.hstack((arr1,arr2))) #hstack函数横向组合 137 138 # 代码 2-25 139 print('纵向组合为:',np.vstack((arr1,arr2))) #vstack函数纵向组合 140 141 # 代码 2-26 142 print('横向组合为:',np.concatenate((arr1,arr2),axis = 1)) #concatenate函数横向组合 143 print('纵向组合为:',np.concatenate((arr1,arr2),axis = 0)) #concatenate函数纵向组合 144 145 # 代码 2-27 146 arr = np.arange(16).reshape(4,4) 147 print('创建的二维数组为:',arr) 148 print('横向分割为:',np.hsplit(arr, 2)) #hsplit函数横向分割 149 150 # 代码 2-28 151 print('纵向分割为:',np.vsplit(arr, 2)) #vsplit函数纵向分割 152 153 # 代码 2-29 154 ''' 155 print('横向分割为:',np.split(arr, 2, axis=1)) #split函数横向分割 156 print('纵向分割为:',np.split(arr, 2, axis=0)) #split函数纵向分割 157 '''