numpy [...]語法簡單使用
Python numpy中切片功能與列表切片類似,但功能更加強大
本文主講numpy中[...]的簡單使用,后續工作繼續補充。
import numpy
>>> a = numpy.array([[1,2,3,4,5],[6,7,8,9,10],[1,2,3,4,5],[6,7,8,9,10]])
>>> a
array([[ 1, 2, 3, 4, 5],
[ 6, 7, 8, 9, 10],
[ 1, 2, 3, 4, 5],
[ 6, 7, 8, 9, 10]])
>>> a[...,2] #表示遍歷每行,2表示索引為2的所在列
array([3, 8, 3, 8])
>>> a[...,:2]#表示遍歷每行,:2表示索引為<2的0,1所在的列
array([[1, 2],
[6, 7],
[1, 2],
[6, 7]])
>>> a[...,::2]#表示遍歷每行,2表示步長,選取多索引為0,2,4所在的列
array([[ 1, 3, 5],
[ 6, 8, 10],
[ 1, 3, 5],
[ 6, 8, 10]])
a[None,...]#相當於插入維度,也想當於reshape(a,[1,4,4])
array([[[[ 1, 2, 3, 4, 5],
[ 6, 7, 8, 9, 10],
[ 1, 2, 3, 4, 5],
[ 6, 7, 8, 9, 10]]]])