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

你有两个矩阵（为了使用矩阵的求和，它必须是IDENTICAL IN SIZE）。只要你让用户定义尺寸（m和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]]

使用python添加Numpy随机矩阵

2 个答案: