Строка Python от мажора до вектора старшего порядка столбца

Имея матрицу как

ma = [[0.343, 0.351, 0.306], [0.145, 0.368, 0.487]]

Я хочу получить такой вектор:

[0.343, 0.145, 0.351, 0.368, 0.306, 0.487]

Чтобы попытаться получить это, я использую numpy а также reshape Но это не работает.

a = np.array(ma)
>>> print a.shape
(2, 3)

Но я получаю:

c = a.reshape(3, 2, order='F')
>>> print c
array([[ 0.343,  0.368],
       [ 0.145,  0.306],
       [ 0.351,  0.487]])

Каков наилучший способ сделать это для любого размера матрицы? Я имею в виду, например, если матрица не в квадрате, как:

[[0.404, 0.571, 0.025],
 [0.076, 0.694, 0.230],
 [0.606, 0.333, 0.061],
 [0.595, 0.267, 0.138]]

Я хотел бы получить:

[0.404, 0.076, 0.606, 0.595, 0.571, 0.694, 0.333, 0.267, 0.025, 0.230, 0.061, 0.138]

4 ответа

Решение

Ты можешь использовать ravel() чтобы сгладить массив.

>>> a.T.ravel()
array([ 0.343,  0.145,  0.351,  0.368,  0.306,  0.487])

# Or specify Fortran order.
>>> a.ravel('F')
array([ 0.343,  0.145,  0.351,  0.368,  0.306,  0.487])

a = np.random.rand(4,2)
>>> a
array([[ 0.59507926,  0.25011282],
       [ 0.68171766,  0.41653172],
       [ 0.83888691,  0.22479481],
       [ 0.04540208,  0.23490886]])

>>> a.T.ravel()  # or a.ravel('F')
array([ 0.59507926,  0.68171766,  0.83888691,  0.04540208,  0.25011282,
        0.41653172,  0.22479481,  0.23490886])

Вы можете получить желаемое, переставив матрицу, а затем используя функцию ravel от numpy:

mat = np.random.rand(3,2)
print np.ravel(mat.T)

Свести массив в порядке Фортрана:

c = a.flatten(order='F')

Вы также можете получить результаты, которые вы хотели с reshape, но это более многословно:

c = a.reshape(a.size, order='F')
>>> A = ([[0, 1, 2],
...        [3, 4, 5],
...        [6, 7, 8]])
>>> 
>>> print(A)
[[0, 1, 2], [3, 4, 5], [6, 7, 8]]
>>> 
>>> [y  for x in [[row[i] for row in A] for i in range(len(A[0]))] for y in x]
[0, 3, 6, 1, 4, 7, 2, 5, 8]
>>> 
>>> 
Другие вопросы по тегам