numpy数组的索引

numpy数组的索引

个人学习笔记，分享给大家！

一、索引的作用

  获取、修改数组单个元素的值

 

二、获取单个元素的值

1，在一维数组中：

import numpy as np 
x = np.array([5,3,6,9,8]) 

x[0] #索引为非负值，获取第i个值，从0开始计数
Out[8]:5 

x[-1] #索引为负值，从末尾开始索引，倒数第一个索引为-1 
Out[9]:8

2，在多维数组中：

import numpy as np 
x = np.eye(3)
 
x 
Out[10]:
array([[1., 0., 0.], [0., 1., 0.], [0., 0., 1.]])

x[0,1] #用逗号分隔的索引元组获取元素 
Out[11]:0.0

三、修改单个元素的值

上接前面的例子：

x[0,1]=5 #通过索引修改数组单个元素的值  

x  
Out[13]:  
array([[1., 5., 0.],  
       [0., 1., 0.],  
       [0., 0., 1.]])