Why does indexing numpy arrays with brackets and commas differ in behavior?

后端 未结 3 832
梦如初夏
梦如初夏 2020-11-27 16:42

I tend to index numpy arrays (matrices) with brackets, but I\'ve noticed when I want to slice an array (matrix) I must use the comma notation. Why is this? For example,

相关标签:
3条回答
  • 2020-11-27 17:13

    This:

    x[:, 1]
    

    means "take all indices of x along the first axis, but only index 1 along the second".

    This:

    x[:][1]
    

    means "take all indices of x along the first axis (so all of x), then take index 1 along the first axis of the result". You're applying the 1 to the wrong axis.

    x[1][2] and x[1, 2] are only equivalent because indexing an array with an integer shifts all remaining axes towards the front of the shape, so the first axis of x[1] is the second axis of x. This doesn't generalize at all; you should almost always use commas instead of multiple indexing steps.

    0 讨论(0)
  • 2020-11-27 17:16

    The real explanation to this is in the case of 2 brackets [][], the first bracket (x[] ) creates an temporary array referring to the first dimension of the multidimensional array.

    While the second bracket of the x[][] would be applied to the already created temporary array.

    So no real hazards are happening. Because of the described behavior, when you combine slicing with multiple brackets basically the first one is already selecting part of the numpy array while the second bracket goes into the already selected array.

    Explanation is available also here : numpy.org/devdocs/user/basics.indexing.html

    0 讨论(0)
  • 2020-11-27 17:18

    When you slice multi dimension of array, if fewer indices are provided than the number of axes, the missing indices are considered complete slices. Hence, when you are essentially doing when calling x[:][1] is x[:,:][1,:] Therefore, x[:,:] will just return x itself.

    0 讨论(0)
提交回复
热议问题