NumPy: iterate over outer dimension of numpy array using nditer

前端 未结 2 1274
盖世英雄少女心
盖世英雄少女心 2021-01-15 00:57

I am unable to iterate over the outer axis of a numpy array.

import numpy as np

a = np.arange(2*3).reshape(2,3)
it = np.nditer(a)
for i in it:
    print i
<         


        
2条回答
  •  难免孤独
    2021-01-15 01:39

    You can iterate it over just like you iterate over a 1D array to get the output like you want.

     for k,v in enumerate(a):
          print(v)
    

提交回复
热议问题