Сложите столбцы в массиве 2-dim, чтобы получить массив 1-dim

Question

Как сложить все столбцы в двумерном массиве Numpy в одномерный массив.

т.е. У меня есть:

x = np.array([[1, 3, 5],[2, 4, 6]])

И я хочу получить:

np.array([1, 2, 3, 4, 5, 6])

Есть ли способ достичь этого без понимания цикла или списка?

Answer 1

Использование flatten с 'F'

x.flatten('F')
Out[114]: array([1, 2, 3, 4, 5, 6])

Answer 2

Вы можете использовать ravel:

x = np.array([[1, 3, 5],[2, 4, 6]])

res = x.ravel('F')  # or x.T.ravel()

# array([1, 2, 3, 4, 5, 6])