《Python数据分析》一2.3 选择NumPy数组元素

2.3 选择NumPy数组元素

有时，我们可能想从数组中选择指定的元素。如何做到这一点呢？不妨从创建一个 2×2矩阵着手（以下代码取自本书代码包中的elementselection.py文件）：

In: a = array([[1,2],[3,4]])
In: a
Out:
array([[1, 2],
　　　　 [3, 4]])

上面的矩阵是通过向array()函数传递一个由列表组成的列表得到的。接下来，我们要逐个选择矩阵的各个元素，代码如下所示。别忘了，下标是从0开始的：

In: a[0,0]
Out: 1
In: a[0,1]
Out: 2
In: a[1,0]
Out: 3
In: a[1,1]
Out: 4

可见，选择数组元素是一件非常简单的事情。对于数组a，只要通过a[m,n]的形式，就能访问数组内的元素，其中m和n为数组元素的下标。数组元素的下标如下所示。