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

[亡魂溺海] 提交于 2019-11-26 07:37:32

问题


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,

>>> x = numpy.array([[1, 2], [3, 4], [5, 6]])
>>> x
array([[1, 2],
       [3, 4],
       [5, 6]])
>>> x[1][1]
4                 # expected behavior
>>> x[1,1]
4                 # expected behavior
>>> x[:][1]
array([3, 4])     # huh?
>>> x[:,1]
array([2, 4, 6])  # expected behavior

回答1:


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.




回答2:


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.




回答3:


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



来源:https://stackoverflow.com/questions/38113994/why-does-indexing-numpy-arrays-with-brackets-and-commas-differ-in-behavior

易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!