ChildishJack - 1 year ago 73

Python Question

I recently posted a question here which was answered exactly as I asked. However, I think I overestimated my ability to manipulate the answer further. I read the broadcasting doc, and followed a few links that led me way back to 2002 about numpy broadcasting.

I've used the second method of array creation using broadcasting:

`N = 10`

out = np.zeros((N**3,4),dtype=int)

out[:,:3] = (np.arange(N**3)[:,None]/[N**2,N,1])%N

which outputs:

`[[0,0,0,0]`

[0,0,1,0]

...

[0,1,0,0]

[0,1,1,0]

...

[9,9,8,0]

[9,9,9,0]]

but I do not understand via the docs how to manipulate that. I would ideally like to be able to set the increments in which each individual column changes.

ex. Column A changes by 0.5 up to 2, column B changes by 0.2 up to 1, and column C changes by 1 up to 10.

`[[0,0,0,0]`

[0,0,1,0]

...

[0,0,9,0]

[0,0.2,0,0]

...

[0,0.8,9,0]

[0.5,0,0,0]

...

[1.5,0.8,9,0]]

Thanks for any help.

Answer Source

You can adjust your current code just a little bit to make it work.

```
>>> out = np.zeros((4*5*10,4))
>>> out[:,:3] = (np.arange(4*5*10)[:,None]//(5*10, 10, 1)*(0.5, 0.2, 1)%(2, 1, 10))
>>> out
array([[ 0. , 0. , 0. , 0. ],
[ 0. , 0. , 1. , 0. ],
[ 0. , 0. , 2. , 0. ],
...
[ 0. , 0. , 8. , 0. ],
[ 0. , 0. , 9. , 0. ],
[ 0. , 0.2, 0. , 0. ],
...
[ 0. , 0.8, 9. , 0. ],
[ 0.5, 0. , 0. , 0. ],
...
[ 1.5, 0.8, 9. , 0. ]])
```

The changes are:

- No
`int`

dtype on the array, since we need it to hold floats in some columns. You could specify a`float`

dtype if you want (or even something more complicated that only allows floats in the first two columns). - Rather than
`N**3`

total values, figure out the number of distinct values for each column, and multiply them together to get our total size. This is used for both`zeros`

and`arange`

. - Use the floor division
`//`

operator in the first broadcast operation because we want integers at this point, but later we'll want floats. - The values to divide by are again based on the number of values for the later columns (e.g. for
`A,B,C`

numbers of values, divide by`B*C, C, 1`

). - Add a new broadcast operation to multiply by various scale factors (how much each value increases at once).
- Change the values in the broadcast mod
`%`

operation to match the bounds on each column.