05 ndarray 切片

最新推荐文章于 2025-05-08 14:51:23 发布

原创

最新推荐文章于 2025-05-08 14:51:23 发布 · 6.9k 阅读

10 ·

CC 4.0 BY-SA版权

这篇博客介绍了NumPy中的切片方法，包括选择子集的三种方式，并讲解了copy方法如何复制数组，以及diag函数如何获取对角线元素。此外，还提到了unique函数用于获取数组中唯一元素的功能。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

一、切片方法

我们除了能够一次访问一个元素之外，NumPy 还提供了访问 ndarray 子集的方式，称之为切片。切片方式是在方括号里用冒号 : 分隔起始和结束索引。通常，你将遇到三种类型的切片：

1. ndarray[start:end]
2. ndarray[start:]
3. ndarray[:end]

第一种方法用于选择在start和end索引之间的元素。第二种方法用于选择从start索引开始直到最后一个索引的所有元素。第三种方法用于选择从第一个索引开始直到 end 索引的所有元素。请注意，在第一种方法和第三种方法中，结束索引不包括在内。此外注意，因为 ndarray可以是多维数组，在进行切片时，通常需要为数组的每个维度指定一个切片。

import numpy as np

# 创建一个5行5列的二维数组
arr = np.arange(25).reshape(5, 5)
print('arr = {}\n'.format(arr))

# 选取arr从第2行(含)到第5行(不含)，第1列(含)到第4列(不含) 组成的ndarray
A = arr[1:4, 0:3]
print('A = {}\n'.format(A))

# 选取arr从第2行(含)到最后一行(含)，第1列(含)到第4列(不含) 组成的ndarray
B = arr[1:, 0:3]
print('B = {}\n'.format(B))

#选取arr从第1行(含)到4行(不含)，第3列(含)到第4列(不含) 组成的ndarray