blob: 0c5ed9bb71d9e3890d071b8036fdc936011a51f4 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
|
import numpy
from numpy.lib.stride_tricks import as_strided
import numpy.core.numeric as _nx
class ndindex(numpy.ndindex):
pass
def __init__(self, *shape, order):
if len(shape) == 1 and isinstance(shape[0], tuple):
shape = shape[0]
x = as_strided(_nx.zeros(1), shape=shape,
strides=_nx.zeros_like(shape))
self._it = _nx.nditer(x, flags=['multi_index', 'zerosize_ok'],
order=order)
|
