使用Python添加随机矩阵与Numpy

Question

所以我需要在Python中添加随机矩阵。我有这个代码来生成一个随机矩阵：

import numpy as np 
import random 

def generate_matrix(m, n): 
    A = np.random.randint(100, size=(m, n)) 
    B = np.random.randint(100, size=(m, n)) 
    return A 

但是，我需要创建一个函数，以便我可以添加两个随机矩阵。 我所拥有的就是启动功能，我只是卡住了。事情是用户需要能够创建每个矩阵的大小。我试图做一个代码，但是我所能想到的是一个代码，其中两个矩阵是相同的，这不是我所需要的。

def add(A, B): 

Answer 1

def generate_matrix(m, n):
A = np.random.randint(100, size=(m, n))
B = np.random.randint(100, size=(m, n))
return A + B
这是我，但我的格式是错误的，它只会增加相同长度的矩阵。

Answer 2

你有两个矩阵（这必须是相同的大小，以便使用矩阵的总和）。 只要你有用户定义的大小（米和n），你很好。

import numpy as np 
import random 

def generate_matrix(m, n): 
    A = np.random.randint(100, size=(m, n)) 
    return(A) 

def user_define(m, n): 
    A = generate_matrix(m, n)  
    B = generate_matrix(m, n) 
    print("Random Matrix 1 is:") 
    print(A) 
    print("Random Matrix 2 is:") 
    print(B) 
    print("Sum of two random matrices is:") 
    print(A + B) 
    return(A + B) 

x = user_define(3, 4) 


Random Matrix 1 is: 
[[47 24 2 46] 
[65 54 3 80] 
[90 5 42 35]] 
Random Matrix 2 is: 
[[80 69 8 50] 
[ 2 71 65 36] 
[32 6 49 6]] 
Sum of two random matrices is: 
[[127 93 10 96] 
[ 67 125 68 116] 
[122 11 91 41]]