Как вернуть массив NumPy с m строками из n случайных целочисленных значений от 0 до 99 в Python? - PullRequest
0 голосов
/ 03 декабря 2018

Я пытаюсь вернуть массив, который принимает m, строки, из n, случайное целое число n, однако я не уверен, как это сделать.Нужно ли устанавливать m = для случайного ввода пользователя?И n = случайное целое число 0-99.Это код, который у меня есть, однако вместо того, чтобы указывать массиву 6 строк из 7 случайных чисел, как мне заменить его на m и n и заставить его работать с использованием случайного числа строк и целых чисел?

import numpy as np
def gen_matrix(m,n):
m= 
return np.random.rand(6,7)*100
random_matrix = numpy.random.randint(0,max_val, 
(<num_rows>,<num_cols>))

1 Ответ

0 голосов
/ 03 декабря 2018

Вы можете использовать следующее:

import numpy as np

def gen_matrix(m,n):
    matrix = np.random.randint(0,99, size=(m, n))
    return matrix

Проверьте это:

my_matrix = gen_matrix(5,5)

print(my_matrix)
...