python基础练习—— 数据分析包numpy数组操作

Posted 2022-06-01 kwu_ganymede

#1. 创建一个2*2的数组，计算对角线上元素的和
import numpy as np
a = np.arange(4).reshape(2,2)
n1 = a[0,1]
n2 = a[1,0]
sum = n1 + n2
print sum

#2. 创建一个长度为9的一维数据，数组元素0到8。将它重新变为3*3的二维数组
import numpy as np
a = np.arange(9).reshape(3,3)
print a

#3. 创建两个3*3的数组，分别将它们合并为3*6、6*3的数组后，拆分为3个数组（维数不限定）
import numpy as np
a = np.arange(9).reshape(3,3)
b = 2 * a

c = np.hstack((a, b))
print c

d = np.vstack((a, b))
print d

print np.hsplit(c, 3)
print np.vsplit(d, 3)

NumPy数组在数值运算方面的效率优于Python提供的list容器。使用NumPy可以在代码中省去很多循环语句，因此其代码比等价的Python代码更为简洁。