Python机器学习手册：从数据预处理到深度学习

1.4

　选择元素

｜

5

1.4

　选择元素

问题描述

在向量或矩阵中选择一个或多个元素。

解决方案

使用

NumPy

的数组来解决这个问题就很简单 ：

#

加载库

import numpy as np

#

创建一个行向量

vector = np.array([1, 2, 3, 4, 5, 6])

#

创建矩阵

matrix = np.array([[1, 2, 3],

[4, 5, 6],

[7, 8, 9]])

#

选择向量的第三个元素

vector[2]

3

#

选择第二行第二列

matrix[1,1]

5

讨论

如同

Python

中的大多数事物一样，

NumPy

数组的索引编号是从

0

开始的，这意味着第

一个元素的下标是

0

而不是

1

。此外，

NumPy

还提供了很多方式来选取元素或数组中的

一组元素（即索引和切片）：

#

选取一个向量的所有元素

vector[:]

array([1, 2, 3, 4, 5, 6])

#

选取从

0

开始一直到第

3

个（包括第

3

个）元素

vector[:3]