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

xiaoxiao2024-03-16 130

本节书摘来自异步社区《Python数据分析》一书中的第2章，第2.3节，作者【印尼】Ivan Idris，更多章节内容可以访问云栖社区“异步社区”公众号查看

有时，我们可能想从数组中选择指定的元素。如何做到这一点呢？不妨从创建一个 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为数组元素的下标。数组元素的下标如下所示。