Python numpy小练习

把《统计建模与R软件》中的一点课后题，用Python中的numpy实现了一下

import numpy as np
#将1,2...,20，组成4x5的矩阵，A为按列输入，B为按行输入
A=np.mat(np.arange(1,21)).reshape(5,4).T
B=np.mat(np.arange(1,21)).reshape(4,5)
print('A:\n',A)
print('B:\n',B)
#计算A+B
print('A+B:\n',A+B)
#计算A*BT
print('A*BT:\n',A@(B.T))
# Eij=(eij)mxn,其中eij=aij*bij
E=np.multiply(A,B)
print('E:\n',E)
#F是A中前三行和前三列构成的矩阵
F=A[0:3,0:3]
print('F:\n',F)
#G是B中各列构成的矩阵，但不包括第三列
G=np.hstack((B[:,0:2],B[:,3:]))#水平拼接hstack，垂直拼接vstack
print('G:\n',G)

运行结果如下：

</