NumPy | put_along_axis method
Start your free 7-days trial now!
put_along_axis(~) sets a specific value in the input array. This is done in-place, that is, no new Numpy array is created.
The input array. All input arrays are treated as a flattened array.
The indices in the specified axis where the values will be set. For 2D arrays, indices would represent the column/row index. The dimensions must be the same as that of
The values to set. If
values is shorter than
values will be repeated to ensure the shape matches up.
The axis along which to insert the values. For 2D arrays, the allowed values are as follows:
Values are set row-wise.
Values are set column-wise.
None - the setting is done in-place.
Consider the following 2D array:
aarray([[5, 6],[7, 8]])
Setting values row-wise
axis=0, like so:
Notice how the dimensions of the indices matches that of the input array
Setting values column-wise
axis=1, like so:
Here, note how
[1,0] does not represent the 2nd row 1st column. Instead, it represents the 1st and 0th index of the flattened version of a. Therefore, we could have equivalently used