numpy 二维数组删除指定值所在的行|极客教程

numpy 二维数组删除指定值所在的行

在进行数据处理和分析时，经常会出现需要删除二维数组中包含特定值的行的情况。在Python中，可以使用NumPy库来处理二维数组的操作。本文将详细介绍如何使用NumPy来删除包含指定值的行。

1. 创建示例二维数组

首先，让我们创建一个示例的二维数组来演示如何删除包含指定值的行。我们使用NumPy的np.array()函数来创建数组。

import numpy as np

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

print("原始数组：")
print(array)

运行以上代码，我们得到了以下示例二维数组：

原始数组：
[[ 1  2  3]
 [ 4  5  6]
 [ 7  8  9]
 [10 11 12]
 [13 14 15]]

2. 删除包含指定值的行

接下来，我们将演示如何使用NumPy来删除包含指定值的行。我们首先定义要删除的指定值。

# 定义要删除的指定值
target_value = 7

然后，我们可以使用NumPy的条件索引功能来操作数组，找到包含指定值的行，并将其删除。

# 使用条件索引删除包含指定值的行
filtered_array = array[array[:, 0] != target_value]

print("\n删除指定值所在行后的数组：")
print(filtered_array)

运行以上代码，我们得到了删除包含指定值的行后的二维数组：

删除指定值所在行后的数组：
[[ 1   2   3]
 [ 4   5  6]
 [10 11 12]
 [13 14 15]]

3. 完整代码

下面是完整的代码示例：

import numpy as np

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

print("原始数组：")
print(array)

# 定义要删除的指定值
target_value = 7

# 使用条件索引删除包含指定值的行
filtered_array = array[array[:, 0] != target_value]

print("\n删除指定值所在行后的数组：")
print(filtered_array)