Я знаю, что простой способ создать массив NxN с нулями в Python:
[[0]*N for x in range(N)]
Однако предположим, что я хочу создать массив, заполнив его случайными числами:
[[random.random()]*N for x in range(N)]
Это не работает, потому что каждый произвольный номер, который создается, затем реплицируется N раз, поэтому мой массив не имеет уникальных случайных чисел NxN.
Есть ли способ сделать это в одной строке, не используя для циклов?
Вы можете использовать понимание вложенного списка:
>>> N = 5
>>> import random
>>> [[random.random() for i in range(N)] for j in range(N)]
[[0.9520388778975947, 0.29456222450756675, 0.33025941906885714, 0.6154639550493386, 0.11409250305307261], [0.6149070141685593, 0.3579148659939374, 0.031188652624532298, 0.4607597656919963, 0.2523207155544883], [0.6372935479559158, 0.32063181293207754, 0.700897108426278, 0.822287873035571, 0.7721460935656276], [0.31035121801363097, 0.2691153671697625, 0.1185063432179293, 0.14822226436085928, 0.5490604341460457], [0.9650509333411779, 0.7795665950184245, 0.5778752066273084, 0.3868760955504583, 0.5364495147637446]]
Или используйте numpy (не-stdlib, но очень популярны):
>>> import numpy as np
>>> np.random.random((N,N))
array([[ 0.26045197, 0.66184973, 0.79957904, 0.82613958, 0.39644677],
[ 0.09284838, 0.59098542, 0.13045167, 0.06170584, 0.01265676],
[ 0.16456109, 0.87820099, 0.79891448, 0.02966868, 0.27810629],
[ 0.03037986, 0.31481138, 0.06477025, 0.37205248, 0.59648463],
[ 0.08084797, 0.10305354, 0.72488268, 0.30258304, 0.230913 ]])
(PS Это хорошая идея, чтобы привыкнуть говорить list, когда вы имеете в виду list и резервируете array для numpy ndarray s. На самом деле есть встроенный модуль array со своим собственным array, так что это путает вещи еще больше, но это относительно редко используется.)
Просто используйте [random.random() for i in range(N)] внутри вашего понимания списка.
Демо:
>>> import random
>>> N = 3
>>> [random.random() for i in range(N)]
[0.24578599816668256, 0.34567935734766164, 0.6482845150243465]
>>> M = 3
>>> [[random.random() for i in range(N)] for j in range(M)]
[[0.9883394519621589, 0.6533595743059281, 0.866522328922242], [0.5906410405671291, 0.4429977939796209, 0.9472377762689498], [0.6883677407216132, 0.8215813727822125, 0.9770711299473647]]
Вы можете использовать списки.
[[random.random() for x in xrange(N)] for y in xrange(N)]
https://docs.python.org/2/tutorial/datastructures.html#list-comprehensions
Для больших многомерных массивов я предлагаю вам использовать numpy.
Это можно сделать без циклы. Попробуйте эту простую строку кода для генерации матрицы случайных чисел 2 на 3 со средним 0 и стандартным отклонением 1.
Синтаксис:
import numpy
numpy.random.normal(mean, standard deviation, (rows,columns))
пример:
numpy.random.normal(0,1,(2,3))