diff options
author | Adrian Kummerlaender | 2019-10-28 22:00:05 +0100 |
---|---|---|
committer | Adrian Kummerlaender | 2019-10-28 22:00:05 +0100 |
commit | 72e7097073b03e6584f603b905b5a4fc236d7def (patch) | |
tree | adfad6f0c9c771dc9cee4b1a441f5ec75904ac46 | |
parent | 091811762b7f2cbb6575139276ea16bf54d3456b (diff) | |
download | boltzgen-72e7097073b03e6584f603b905b5a4fc236d7def.tar boltzgen-72e7097073b03e6584f603b905b5a4fc236d7def.tar.gz boltzgen-72e7097073b03e6584f603b905b5a4fc236d7def.tar.bz2 boltzgen-72e7097073b03e6584f603b905b5a4fc236d7def.tar.lz boltzgen-72e7097073b03e6584f603b905b5a4fc236d7def.tar.xz boltzgen-72e7097073b03e6584f603b905b5a4fc236d7def.tar.zst boltzgen-72e7097073b03e6584f603b905b5a4fc236d7def.zip |
Use order-accepting ndindex method for generating cell indices
-rw-r--r-- | boltzgen/geometry.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/boltzgen/geometry.py b/boltzgen/geometry.py index 7cc406d..d4411ad 100644 --- a/boltzgen/geometry.py +++ b/boltzgen/geometry.py @@ -24,8 +24,8 @@ class Geometry: else: raise Exception('Geometry malformed') - def inner_cells(self): - for idx in numpy.ndindex(self.inner_size()): + def inner_cells(self, order='F'): + for idx in ndindex(self.inner_size(), order): yield tuple(map(lambda i: i + 1, idx)) def size(self): |