NumPy支持类似列表的定位操作,示例如下。
import numpy as np matrix = np.array([[1,2,3],[20,30,40]]) print(matrix[0,1])
得到的结果是2。
上述代码中,matrix[0,1]中0代表的是行,在NumPy中,0代表起始的第一个,因此取的是第1行,之后的1代表列,因此取的是第2列。那么,最后第一行第二列就是2这个值了。