Как выполнить итерацию массива np.zeros в python, печатая столбцы один за другим?

Question

У меня есть массив np.zeros, который был добавлен в другой цикл, который теперь выглядит как

array([[ 0.,  0.],
   [ 4.,  4.],
   [ 0.,  0.],
   [ 0.,  0.],
   [ 0.,  0.],
   [ 0.,  0.],
   [ 0.,  0.],
   [ 3.,  3.],
   [ 0.,  3.],
   [ 0.,  0.],
   [ 0.,  0.],
   [ 0.,  0.],
   [ 0.,  0.],
   [ 0.,  0.],
   [ 0.,  0.],
   [ 0.,  0.],
   [ 0.,  0.],
   [ 0.,  0.],
   [ 0.,  0.],
   [ 0.,  0.]])

Как мне создать цикл, который печатает первый столбец, а затем следующий столбец?

Я попробовал следующий код,

for i in goalsfor[:,i]:
    print(goalsfor)

но возникает следующая ошибка.

IndexError: index 19 is out of bounds for axis 1 with size 2

Кроме того, можно ли добавить массив np.zero к столбцу?

Спасибо!

Answer 1

Ваш отпечаток:

for i in range(goalsfor.shape[1]):
    print(goalsfor[:,i])

и вы можете добавить по столбцу с np.column_stack