python任务2

python任务2：

• （1）生成矩阵
  $$A=\left[\begin{array}{ccccc}1& 6& 4& 3& 0\\ 4& 1& 3& 0& 0\\ 5& 0& 0& 1& 0\\ 0& 0& 0& 0& 0\\ 1& 1& 3& 6& 8\end{array}\right]B=\left[\begin{array}{ccccc}1& 2& 4& 7& 0\\ 0& 1& 1& 8& 0\\ 2& 4& 0& 9& 0\\ 1& 2& 6& 5& 8\\ 2& 4& 0& 2& 1\end{array}\right]$$
  $$A_{ij}=R^{5\delta }i,j\in [1,5]B_{ij}=R^{55}i,j\in [1,5]$$
• （2）分别计算出A矩阵与B矩阵每行元素的和。
• （3）分别计算出A矩阵与B矩阵每列元素的和。
• （4）分别计算出A矩阵和B矩阵所有元素的和。
• （5） 计算A矩阵元素除以该A矩阵所有元素和，输出得到的矩阵存入名为“normalization.txt”文件中。
• （6）计算矩阵A乘以矩阵B，输出所得矩阵的形状并将矩阵存入名为“multiply.txt”文件中。
• （7）输出矩阵A的转置
• （8）计算矩阵A加上矩阵B的和。

import numpy as np

# 创建矩阵A和矩阵B
a = np.array([
    [1, 6, 4, 3, 0],
    [4, 1, 3, 0, 0],
    [5, 0, 0, 1, 0],
    [0, 0, 0, 0, 0,],
    [1, 1, 3, 6, 8]
])
b = np.array([
    [1, 2, 4, 7, 0],
    [0, 1, 1, 8, 0],
    [2, 4, 0, 9, 0],
    [1, 2, 6, 5, 8],
    [2, 4, 0, 2, 1]
])
print(a)
print(b)

[[1 6 4 3 0]
 [4 1 3 0 0]
 [5 0 0 1 0]
 [0 0 0 0 0]
 [1 1 3 6 8]]
[[1 2 4 7 0]
 [0 1 1 8 0]
 [2 4 0 9 0]
 [1 2 6 5 8]
 [2 4 0 2 1]]

# 矩阵A和矩阵B每行元素的和
print(a.sum(axis=1)) 
print(b.sum(axis=1))

[14  8  6  0 19]
[14 10 15 22  9]

# 矩阵A和矩阵B每列元素的和
print(a.sum(axis=0)) 
print(b.sum(axis=0))

[11  8 10 10  8]
[ 6 13 11 31  9]

#矩阵A和矩阵B所有元素的和
print(a.sum())
print(b.sum())

47
70

# 计算A矩阵元素除以该A矩阵所有元素和，输出得到的矩阵存入名为“normalization.txt”文件中
f = open('normalization.txt',"w")
normal = a/a.sum()
print(normal, file=f,flush=True)

# 计算矩阵A乘以矩阵B，输出所得矩阵的形状并将矩阵存入名为“multiply.txt”文件中
f_multi = open('multiply.txt',"w")
multi = a * b
print(multi, file=f_multi,flush=True)

# 输出矩阵A的转置
print(a.T)

[[1 4 5 0 1]
 [6 1 0 0 1]
 [4 3 0 0 3]
 [3 0 1 0 6]
 [0 0 0 0 8]]

# 计算矩阵A加上矩阵B的和
sum = a + b
print(sum)

[[ 2  8  8 10  0]
 [ 4  2  4  8  0]
 [ 7  4  0 10  0]
 [ 1  2  6  5  8]
 [ 3  5  3  8  9]]