python获取多个指定行(列)的值

只要用元组就可以啦,如下面(0,3)表示获取所有行第0列和第3列的值。

>>>import numpy as np
>>>array = np.arange(20).reshape(4, 5)
>>>array
   array([[ 0,  1,  2,  3,  4],
          [ 5,  6,  7,  8,  9],
          [10, 11, 12, 13, 14],
          [15, 16, 17, 18, 19]])
>>> print array[:, (0, 3)]
   array([[ 0,  3],
          [ 5,  8],
          [10, 13],
          [15, 18]])


版权声明:本文为csdnliwenqi原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。