Update README.md
Signed-off-by: David Rotermund <54365609+davrot@users.noreply.github.com>
This commit is contained in:
parent
6206f2a3bf
commit
37d3f569f8
1 changed files with 103 additions and 0 deletions
|
@ -739,3 +739,106 @@ a:6 b:2
|
|||
a:7 b:2
|
||||
a:8 b:2
|
||||
```
|
||||
|
||||
Or only one matrix:
|
||||
|
||||
```python
|
||||
import numpy as np
|
||||
|
||||
a = np.arange(0, 9).reshape(3, 3)
|
||||
|
||||
print(a)
|
||||
print()
|
||||
|
||||
|
||||
with np.nditer([a]) as iterator:
|
||||
for x in iterator:
|
||||
print(f"a:{x}")
|
||||
print()
|
||||
|
||||
with np.nditer([a], op_axes=[[0]]) as iterator:
|
||||
for x in iterator:
|
||||
print(f"a:{x}")
|
||||
print()
|
||||
|
||||
with np.nditer([a], op_axes=[[1]]) as iterator:
|
||||
for x in iterator:
|
||||
print(f"a:{x}")
|
||||
print()
|
||||
```
|
||||
|
||||
Output:
|
||||
|
||||
```python
|
||||
[[0 1 2]
|
||||
[3 4 5]
|
||||
[6 7 8]]
|
||||
|
||||
a:0
|
||||
a:1
|
||||
a:2
|
||||
a:3
|
||||
a:4
|
||||
a:5
|
||||
a:6
|
||||
a:7
|
||||
a:8
|
||||
|
||||
a:0
|
||||
a:3
|
||||
a:6
|
||||
|
||||
a:0
|
||||
a:1
|
||||
a:2
|
||||
```
|
||||
|
||||
## parameters galore
|
||||
|
||||
> **flags** : sequence of str, optional
|
||||
>
|
||||
> **delay_bufalloc** delays allocation of the buffers until a **reset()** call is made. Allows allocate operands to be initialized before their values are copied into the buffers.
|
||||
|
||||
> **op_flags** : list of list of str, optional
|
||||
>
|
||||
> **allocate** causes the array to be allocated if it is None in the op parameter.
|
||||
|
||||
```python
|
||||
nditer.reset()
|
||||
```
|
||||
|
||||
> Reset the iterator to its initial state.
|
||||
|
||||
```python
|
||||
import numpy as np
|
||||
|
||||
a = np.arange(0, 9).reshape(3, 3)
|
||||
|
||||
print(a)
|
||||
print()
|
||||
|
||||
with np.nditer(
|
||||
[a, None],
|
||||
flags=["reduce_ok", "buffered", "delay_bufalloc"],
|
||||
op_flags=[["readonly"], ["readwrite", "allocate"]],
|
||||
op_axes=[None, [0, -1]],
|
||||
) as iterator:
|
||||
iterator.operands[1][...] = 0
|
||||
iterator.reset()
|
||||
|
||||
for x, y in iterator:
|
||||
y[...] += x
|
||||
out = iterator.operands[1]
|
||||
print(out)
|
||||
```
|
||||
|
||||
Output:
|
||||
|
||||
```python
|
||||
[[0 1 2]
|
||||
[3 4 5]
|
||||
[6 7 8]]
|
||||
|
||||
[ 3 12 21]
|
||||
```
|
||||
|
||||
|
|
Loading…
Reference in a new issue