本节书摘来自异步社区《Python数据分析》一书中的第2章,第2.3节,作者【印尼】Ivan Idris,更多章节内容可以访问云栖社区“异步社区”公众号查看
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为数组元素的下标。数组元素的下标如下所示。