Numpy 索引: 返回剩余部分

阅读更多：Numpy 教程

简介

在使用Numpy时，索引是非常重要的。Numpy中的索引有一些非常强大的功能。本篇文章将介绍如何使用Numpy索引返回剩余部分。

索引基础

在开始介绍如何返回剩余部分之前，我们需要先了解一些Numpy索引的基础知识。

Numpy索引基础包括两个方面：基本索引和高级索引。

基本索引

基本索引使用整数或者切片来访问数组中的元素。可以使用一个整数或者一个切片，也可以使用一个整数和一个切片的组合。

下面是一些基本索引的例子：

import numpy as np

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

# 使用整数访问数组中的单个元素
print(arr[0])

# 使用切片访问数组中的一段元素
print(arr[1:3])

# 使用整数和切片的组合访问数组中的一段元素
print(arr[0:3:2])

高级索引

高级索引使用整数、列表或者布尔数组来访问数组中的元素。高级索引提供了一些非常强大的功能，可以用来选择数组中的子集。

下面是一些高级索引的例子：

import numpy as np

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

# 使用整数数组访问数组中的元素
idx = np.array([0, 1])
print(arr[idx])

# 使用切片和整数数组访问数组中的元素
idx = np.array([0, 1])
print(arr[1:, idx])

# 使用布尔数组访问数组中的元素
bool_idx = (arr > 2) & (arr < 5)
print(arr[bool_idx])

返回剩余部分

如果我们想要从数组中删除某些元素，并返回剩余的元素，则可以使用Numpy的索引功能。

删除单个元素

我们可以使用基本索引来删除单个元素。

下面是一个例子：

import numpy as np

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

# 删除数组中的第一个元素
res = arr[1:]
print(res)

输出结果为：

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

删除多个元素

我们可以使用高级索引来删除多个元素。

下面是一个例子：

import numpy as np

arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

# 删除第一行和第二列
bool_idx = np.array([[False, True, False], [True, False, True], [False, True, False]])
res = arr[bool_idx]
print(res)

输出结果为：