使用坐标列表高效更新NumPy数组：高级索引技巧解析

本文旨在探讨如何使用坐标列表高效地更新numpy二维数组。我们将分析常见的错误尝试，例如不正确的`dtype`定义和顺序索引，并重点介绍两种正确的numpy高级索引方法：利用二维整数数组和结构化数组。通过代码示例，文章将详细阐述如何利用这些技术实现数组的批量更新，同时强调避免python循环以优化性能。

在数据科学和数值计算中，NumPy数组因其高效的向量化操作而广泛应用。当需要根据一系列坐标来更新NumPy数组中的特定元素时，理解并正确使用NumPy的高级索引机制至关重要。本教程将深入探讨如何以专业且高效的方式实现这一目标。

理解常见的索引误区

假设我们有一个初始化为零的NumPy二维数组，并希望根据一组坐标将其所有元素加一。一个常见的错误尝试是使用Python循环或不当的索引方式。

考虑以下初始代码：

import numpy as np

def update(coords):
    # 这里的coords预期是单个(x, y)元组，但实际传入的是一个坐标数组
    # 这种索引方式对于数组索引是错误的
    return np_arr[coords[0]][coords[1]] + 1

size = 3
np_arr = np.zeros((size, size))
# 尝试定义一个包含两个整数的dtype，但这不是创建元组数组的方式
dt = np.dtype('int', 'int') 
np_indices = np.array([(x, y) for y in range(size) for x in range(size)], dtype=dt)

# 尝试用整个坐标数组进行更新
np_arr = update(np_indices)
print(np_arr)

这段代码的预期输出是一个所有元素都为1的 3x3 矩阵，但实际输出的形状不符，且结果不正确。这背后存在几个关键问题：

1. np.dtype('int', 'int') 的误解： 当使用 np.dtype('int', 'int') 定义 dtype 时，NumPy并不会创建一个包含元组的数组，而是创建一个二维的整数数组。例如，对于 size=3，np_indices 的实际结构是：

   array([[0, 0],
          [1, 0],
          [2, 0],
          [0, 1],
          [1, 1],
          [2, 1],
          [0, 2],
          [1, 2],
          [2, 2]])

   登录后复制

   这是一个 (9, 2) 形状的二维整数数组，而不是一个包含9个 (x, y) 元组的一维数组。

2. 顺序索引的局限性： 表达式 np_arr[coords[0]][coords[1]] 适用于标量索引（如 np_arr[2][3] 等同于 np_arr[2,3]）。但当 coords[0] 和 coords[1] 都是数组时，这种写法会导致问题。np_arr[coords[0]] 会首先执行一次索引操作，返回一个子数组，然后 [coords[1]] 会尝试对这个子数组进行第二次索引。这种顺序索引不适用于同时使用多个数组作为索引来定位二维数组中的多个点。

正确的NumPy高级索引方法

要高效且正确地使用坐标列表更新NumPy数组，我们需要利用NumPy的高级索引（Advanced Indexing）特性。高级索引允许我们使用整数数组或布尔数组来同时索引多个不连续的元素。

方法一：使用二维整数数组进行高级索引

如果我们的坐标列表是一个 (N, 2) 形状的二维整数数组（其中N是坐标点的数量，2代表x和y坐标），我们可以通过将其拆分为两个一维数组来执行高级索引。

import numpy as np

size = 3
np_arr = np.zeros((size, size))

# 创建一个 (N, 2) 形状的整数数组作为坐标列表
# 这里的 dtype='int' 是默认行为，不需要特殊指定
np_indices = np.array([(x, y) for y in range(size) for x in range(size)], dtype=int)

print("原始 np_arr:\n", np_arr)
print("坐标数组 np_indices:\n", np_indices)

# 使用高级索引：将第一列作为行索引，第二列作为列索引
# np_indices[:, 0] 获取所有行的第一列 (x坐标)
# np_indices[:, 1] 获取所有行的第二列 (y坐标)
np_arr[np_indices[:, 0], np_indices[:, 1]] += 1

print("\n更新后的 np_arr:\n", np_arr)

输出：

原始 np_arr:
 [[0. 0. 0.]
 [0. 0. 0.]
 [0. 0. 0.]]
坐标数组 np_indices:
 [[0 0]
 [1 0]
 [2 0]
 [0 1]
 [1 1]
 [2 1]
 [0 2]
 [1 2]
 [2 2]]

更新后的 np_arr:
 [[1. 1. 1.]
 [1. 1. 1.]
 [1. 1. 1.]]

解释：np_arr[np_indices[:, 0], np_indices[:, 1]] 是一种高效的向量化操作。它将 np_indices[:, 0] 中的每个元素作为行索引，并将其对应的 np_indices[:, 1] 中的元素作为列索引，从而同时访问并更新所有指定的 (row, col) 对。

小羊标书

一键生成百页标书，让投标更简单高效

62

查看详情

方法二：使用结构化数组进行高级索引

如果我们确实需要一个包含元组（或具有命名字段的记录）的数组，NumPy的结构化数组（Structured Arrays）是正确的选择。

import numpy as np

size = 3
np_arr = np.zeros((size, size))

# 定义一个包含 'x' 和 'y' 字段的结构化 dtype
dt = np.dtype([('x', 'int'), ('y', 'int')])

# 创建结构化数组
np_indices_structured = np.array([(x, y) for y in range(size) for x in range(size)], dtype=dt)

print("原始 np_arr:\n", np_arr)
print("结构化坐标数组 np_indices_structured:\n", np_indices_structured)
print("访问 x 坐标: ", np_indices_structured['x'])
print("访问 y 坐标: ", np_indices_structured['y'])

# 使用结构化数组的字段进行高级索引
np_arr[np_indices_structured['x'], np_indices_structured['y']] += 1

print("\n更新后的 np_arr:\n", np_arr)

输出：

原始 np_arr:
 [[0. 0. 0.]
 [0. 0. 0.]
 [0. 0. 0.]]
结构化坐标数组 np_indices_structured:
 [(0, 0) (1, 0) (2, 0) (0, 1) (1, 1) (2, 1) (0, 2) (1, 2) (2, 2)]
访问 x 坐标:  [0 1 2 0 1 2 0 1 2]
访问 y 坐标:  [0 0 0 1 1 1 2 2 2]

更新后的 np_arr:
 [[1. 1. 1.]
 [1. 1. 1.]
 [1. 1. 1.]]

解释： 通过 dt = np.dtype([('x', 'int'), ('y', 'int')])，我们创建了一个一维的结构化数组，其中每个元素都是一个记录，包含名为 x 和 y 的字段。我们可以通过字段名（如 np_indices_structured['x']）来访问所有 x 坐标，这同样返回一个一维整数数组，然后可以像方法一那样进行高级索引。

性能考量：避免Python循环

虽然可以通过遍历结构化数组来更新元素，例如：

# 这种方式虽然可行，但效率低下，不推荐用于大型数组
# for x, y in np_indices_structured:
#     np_arr[x, y] += 1

但这种方法在Python层面上进行迭代，对于大型数组而言效率远低于NumPy的向量化高级索引操作。NumPy的设计哲学是尽可能将操作推送到底层的C语言实现，以获得最佳性能。因此，始终优先考虑使用上述两种高级索引方法进行批量更新。

总结

正确地使用NumPy高级索引是进行高效数组操作的关键。在根据坐标列表更新NumPy数组时，应避免以下误区：

• 误用 np.dtype('int', 'int') 期望得到元组数组。
• 对数组索引使用顺序索引 arr[idx1][idx2]。

相反，请采用以下推荐方法：

1. 对于 (N, 2) 形状的整数坐标数组：使用 arr[coords[:, 0], coords[:, 1]] 进行同时索引。
2. 对于结构化坐标数组：使用 arr[coords['x'], coords['y']] （假设字段名为 'x' 和 'y'）进行索引。

始终优先选择向量化的NumPy操作，以确保代码的性能和可扩展性。掌握这些高级索引技巧将使您能够更有效地处理复杂的数组操作任务。

以上就是使用坐标列表高效更新NumPy数组：高级索引技巧解析的详细内容，更多请关注php中文网其它相关文章！