2.3 Numpy数值计算基础

###############################################################################
#######################            正文代码             #######################
###############################################################################

# 代码 2-39
import numpy as np  #导入NumPy库
arr = np.arange(100).reshape(10,10)  #创建一个数组
np.save("../tmp/save_arr",arr)  #保存数组
print('保存的数组为：\n',arr)

# 代码 2-40
arr1 = np.array([[1,2,3],[4,5,6]])
arr2 = np.arange(0,1.0,0.1)
np.savez('../tmp/savez_arr',arr1,arr2)
print('保存的数组1为：',arr1)
print('保存的数组2为：',arr2)

# 代码 2-41
loaded_data = np.load("../tmp/save_arr.npy")  #读取含有单个数组的文件
print('读取的数组为：\n',loaded_data)

loaded_data1 = np.load("../tmp/savez_arr.npz")  #读取含有多个数组的文件
print('读取的数组1为：',loaded_data1['arr_0'])
print('读取的数组2为：',loaded_data1['arr_1'])

# 代码 2-42
arr = np.arange(0,12,0.5).reshape(4,-1)
print('创建的数组为：',arr)

#fmt ="%d"为指定保存为整数
np.savetxt("../tmp/arr.txt", arr, fmt="%d", delimiter=",")
#读入的时候也需要指定逗号分隔
loaded_data = np.loadtxt("../tmp/arr.txt",delimiter=",")
print('读取的数组为：',loaded_data)


# 代码 2-43
loaded_data = np.genfromtxt("../tmp/arr.txt", delimiter = ",")
print('读取的数组为：',loaded_data)

# 代码 2-44
np.random.seed(42)  #设置随机种子
arr = np.random.randint(1,10,size = 10)  #生成随机数
print('创建的数组为：',arr)

arr.sort()  #直接排序
print('排序后数组为：',arr)

arr = np.random.randint(1,10,size = (3,3))  #生成3行3列的随机数
print('创建的数组为：',arr)

arr.sort(axis = 1)  #沿着横轴排序
print('排序后数组为：',arr)

arr.sort(axis = 0)  #沿着纵轴排序
print('排序后数组为：',arr)

# 代码 2-45
arr = np.array([2,3,6,8,0,7])
print('创建的数组为：',arr)
print('排序后数组为：',arr.argsort())  #返回值为重新排序值的下标


# 代码 2-46
a = np.array([3,2,6,4,5])
b = np.array([50,30,40,20,10])
c = np.array([400,300,600,100,200])
d = np.lexsort((a,b,c))  #lexsort函数只接受一个参数，即（a,b,c）
#多个键值排序是按照最后一个传入数据计算的
print('排序后数组为：',list(zip(a[d],b[d],c[d])))


# 代码 2-47
names = np.array(['小明', '小黄', '小花', '小明', 
    '小花', '小兰', '小白'])
print('创建的数组为：',names)

print('去重后的数组为：',np.unique(names))

#跟np.unique等价的Python代码实现过程
print('去重后的数组为：',sorted(set(names)))


ints = np.array([1,2,3,4,4,5,6,6,7,8,8,9,10])  #创建数值型数据
print('创建的数组为：',ints)
print('去重后的数组为：',np.unique(ints))

# 代码 2-48
arr = np.arange(5)
print('创建的数组为：',arr)
print('重复后数组为：',np.tile(arr,3))  #对数组进行重复

# 代码 2-49
np.random.seed(42)  #设置随机种子
arr = np.random.randint(0,10,size = (3,3))
print('创建的数组为：',arr)
print('重复后数组为：',arr.repeat(2, axis = 0))  #按行进行元素重复
print('重复后数组为：',arr.repeat(2, axis = 1))  #按列进行元素重复

# 代码 2-50
arr = np.arange(20).reshape(4,5)
print('创建的数组为：',arr)
print('数组的和为：',np.sum(arr))  #计算数组的和
print('数组横轴的和为：',arr.sum(axis = 0))  #沿着横轴计算求和
print('数组纵轴的和为：',arr.sum(axis = 1))  #沿着纵轴计算求和
print('数组的均值为：',np.mean(arr))  #计算数组均值
print('数组横轴的均值为：',arr.mean(axis = 0))  #沿着横轴计算数组均值
print('数组纵轴的均值为：',arr.mean(axis = 1))  #沿着纵轴计算数组均值
print('数组的标准差为：',np.std(arr))  #计算数组标准差
print('数组的方差为：',np.var(arr))  #计算数组方差
print('数组的最小值为：',np.min(arr))  #计算数组最小值
print('数组的最大值为：',np.max(arr))  #计算数组最大值
print('数组的最小元素为：',np.argmin(arr))  #返回数组最小元素的索引
print('数组的最大元素为：',np.argmax(arr))  #返回数组最大元素的索引

# 代码 2-51
arr = np.arange(2,10)
print('创建的数组为：',arr)
print('数组元素的累计和为：',np.cumsum(arr))  #计算所有元素的累计和
print('数组元素的累计积为：',np.cumprod(arr))  #计算所有元素的累计积


###############################################################################
#######################            任务实现             #######################
###############################################################################

# 代码 2-52
iris_sepal_length = np.loadtxt("../data/iris_sepal_length.csv", 
    delimiter=",")  #读取文件
print('花萼长度表为：',iris_sepal_length)

iris_sepal_length.sort()  #对数据进行排序
print('排序后的花萼长度表为：',iris_sepal_length)

#去除重复值
print('去重后的花萼长度表为：',np.unique(iris_sepal_length))

print('花萼长度表的总和为：',np.sum(iris_sepal_length))  #计算数组总和

#计算所有元素的累计和
print('花萼长度表的累计和为：',np.cumsum(iris_sepal_length))

print('花萼长度表的均值为：',np.mean(iris_sepal_length))  #计算数组均值

#计算数组标准差
print('花萼长度表的标准差为：',np.std(iris_sepal_length))

print('花萼长度表的方差为：',np.var(iris_sepal_length))  #计算数组方差
print('花萼长度表的最小值为：',np.min(iris_sepal_length))  #计算最小值
print('花萼长度表的最大值为：',np.max(iris_sepal_length))  #计算最大值

运行结果：

C:\Python34\python.exe "C:/Users/123/Desktop/数据分析源代码/数据分析源代码/第2章/code/任务2.2 认识NumPy矩阵与通用函数.py"
创建的矩阵为： [[1 2 3]
 [4 5 6]
 [7 8 9]]
创建的矩阵为： [[123]
 [456]
 [789]]
创建的数组1为： [[1. 0. 0.]
 [0. 1. 0.]
 [0. 0. 1.]]
创建的数组2为： [[3. 0. 0.]
 [0. 3. 0.]
 [0. 0. 3.]]
创建的矩阵为： [[1. 0. 0. 3. 0. 0.]
 [0. 1. 0. 0. 3. 0.]
 [0. 0. 1. 0. 0. 3.]
 [1. 0. 0. 3. 0. 0.]
 [0. 1. 0. 0. 3. 0.]
 [0. 0. 1. 0. 0. 3.]]
创建的矩阵为： [[1 2 3]
 [4 5 6]
 [7 8 9]]
创建的矩阵为： [[ 3  6  9]
 [12 15 18]
 [21 24 27]]
矩阵相加结果为： [[ 4  8 12]
 [16 20 24]
 [28 32 36]]
矩阵相减结果为： [[ -2  -4  -6]
 [ -8 -10 -12]
 [-14 -16 -18]]
矩阵相乘结果为： [[ 90 108 126]
 [198 243 288]
 [306 378 450]]
矩阵对应元素相乘结果为： [[  3  12  27]
 [ 48  75 108]
 [147 192 243]]
数组相加结果为： [5 7 9]
数组相减结果为： [-3 -3 -3]
数组相乘结果为： [ 4 10 18]
数组相除结果为： [0.25 0.4  0.5 ]
数组幂运算结果为： [  1  32 729]
数组比较结果为： [ True False False]
数组比较结果为： [False False  True]
数组比较结果为： [False  True False]
数组比较结果为： [False  True  True]
数组比较结果为： [ True  True False]
数组比较结果为： [ True False  True]
数组逻辑运算结果为： False
数组逻辑运算结果为： True
创建的数组1为： [[0 0 0]
 [1 1 1]
 [2 2 2]
 [3 3 3]]
数组1的shape为： (4, 3)
创建的数组2为： [1 2 3]
数组2的shape为： (3,)
数组相加结果为： [[1 2 3]
 [2 3 4]
 [3 4 5]
 [4 5 6]]
创建的数组1为： [[0 0 0]
 [1 1 1]
 [2 2 2]
 [3 3 3]]
数组1的shape为： (4, 3)
创建的数组2为： [[1]
 [2]
 [3]
 [4]]
数组2的shape为： (4, 1)
数组相加结果为： [[1 1 1]
 [3 3 3]
 [5 5 5]
 [7 7 7]]

Process finished with exit code 0