code: def orbit_gen(self):
          
          if self.norbits == 1:
              yield self
  
          else:
              for i in range(self.norbits):
                  yield self[:, i]
docs: Generator for iterating over each orbit.