从Numpy数组中删除最后N行的方法

有时候我们需要从Numpy数组中删除最后N行的数据。本文章将介绍两种常见的方法来实现这个需求。

方法一：使用切片操作

import numpy as np

# 创建一个示例Numpy数组
arr = np.array([[1, 2, 3],
                [4, 5, 6],
                [7, 8, 9],
                [10, 11, 12],
                [13, 14, 15]])

N = 2  # 要删除的行数

# 使用切片删除最后N行
new_arr = arr[:-N]

在这个例子中，我们首先创建了一个5行3列的Numpy数组。然后，我们指定要删除的行数为2。接下来，我们使用切片操作 [:-N] 来删除最后N行。最终，得到的结果将是一个包含了剩余行数的新的Numpy数组。

方法二：使用numpy.delete()函数

import numpy as np

# 创建一个示例Numpy数组
arr = np.array([[1, 2, 3],
                [4, 5, 6],
                [7, 8, 9],
                [10, 11, 12],
                [13, 14, 15]])

N = 2  # 要删除的行数

# 使用numpy.delete()函数删除最后N行
new_arr = np.delete(arr, np.s_[-N:], axis=0)

在这个例子中，我们使用了numpy.delete()函数来删除最后N行。函数的第一个参数是要删除的数组，第二个参数是要删除的行索引，我们使用了np.s_[-N:]来生成一个切片对象，表示从倒数第N行到最后一行，第三个参数axis表示操作的维度，这里我们指定为0表示删除行。最终，得到的结果将是一个包含了剩余行数的新的Numpy数组。

以上是两种常见的方法来从Numpy数组中删除最后N行的方式。你可以根据具体的需求选择其中一种方法来使用。