Efficient way to add a singleton dimension to a NumPy vector so that slice assignments work

前端 未结 5 872
清歌不尽
清歌不尽 2020-12-11 00:48

In NumPy, how can you efficiently make a 1-D object into a 2-D object where the singleton dimension is inferred from the current object (i.e. a list should go to either a 1x

5条回答
  •  Happy的楠姐
    2020-12-11 01:16

    In the most general case, the easiest way to add extra dimensions to an array is by using the keyword None when indexing at the position to add the extra dimension. For example

    my_array = numpy.array([1,2,3,4])
    
    my_array[None, :] # shape 1x4
    
    my_array[:, None] # shape 4x1
    

提交回复
热议问题