Объединить все строки матрицы numpy в python
у меня есть матрица numpy и я хотел бы объединить все строки вместе, поэтому я получаю один длинный массив.
#example
input:
[[1 2 3]
[4 5 6}
[7 8 9]]
output:
[[1 2 3 4 5 6 7 8 9]]
так, как я делаю это сейчас не кажется весть. Я уверен, что есть лучший путь.
combined_x = x[0]
for index, row in enumerate(x):
if index!= 0:
combined_x = np.concatenate((combined_x,x[index]),axis=1)
Спасибо за помощь.
2 ответов
Я бы предложил ravel
или flatten
метод ndarray
.
>>> a = numpy.arange(9).reshape(3, 3)
>>> a.ravel()
array([0, 1, 2, 3, 4, 5, 6, 7, 8])
ravel
быстрее concatenate
и flatten
потому что он не возвращает копию, если это:
>>> a.ravel()[5] = 99
>>> a
array([[ 0, 1, 2],
[ 3, 4, 99],
[ 6, 7, 8]])
>>> a.flatten()[5] = 77
>>> a
array([[ 0, 1, 2],
[ 3, 4, 99],
[ 6, 7, 8]])
но если вам нужна копия, чтобы избежать обмена памятью, показанного выше, вам лучше использовать flatten
чем concatenate
, как вы можете видеть из этих таймингов:
>>> %timeit a.ravel()
1000000 loops, best of 3: 468 ns per loop
>>> %timeit a.flatten()
1000000 loops, best of 3: 1.42 us per loop
>>> %timeit numpy.concatenate(a)
100000 loops, best of 3: 2.26 us per loop
Отметим также, что вы можете достичь точно результат, который иллюстрирует ваш вывод (однорядный 2-d массив) с reshape
(спасибо Pierre GM!):
>>> a = numpy.arange(9).reshape(3, 3)
>>> a.reshape(1, -1)
array([[0, 1, 2, 3, 4, 5, 6, 7, 8]])
>>> %timeit a.reshape(1, -1)
1000000 loops, best of 3: 736 ns per loop
вы можете использовать numpy concatenate
функции:
>>> ar = np.array([[1,2,3],[4,5,6],[7,8,9]])
>>> np.concatenate(ar)
array([1, 2, 3, 4, 5, 6, 7, 8, 9])
вы также можете попробовать flatten
:
>>> ar.flatten()
array([1, 2, 3, 4, 5, 6, 7, 8, 9])