Numpy 数组中的值交换方法

当我们使用Numpy处理数据时，有时需要交换数组中的值。这篇文章将介绍在Numpy中如何交换数组中两个值。

阅读更多：Numpy 教程

Numpy数组

在介绍交换方法之前，先看看Numpy数组的基本特性。

Numpy数组可以是任意维度。我们可以使用numpy模块中的array方法来创建一个数组，并且可以使用dtype参数指定数组类型。

import numpy as np

# 创建一个一维数组
arr1 = np.array([1, 2, 3])

# 创建一个二维数组
arr2 = np.array([[1, 2, 3], [4, 5, 6]])

# 创建一个整型数组
arr3 = np.array([1, 2, 3], dtype=int)

交换数组中的两个值

在Numpy中，我们可以使用fancy indexing（花式索引）和advanced indexing（高级索引）来交换数组中的值。

使用fancy indexing

当我们要交换数组中的两个值时，我们可以使用fancy indexing。先使用numpy.where方法找到要交换的元素的下标，然后通过索引来交换元素。

import numpy as np

# 创建一个一维数组
arr1 = np.array([1, 2, 3])

# 使用where方法找到要交换的元素的下标
idx1 = np.where(arr1 == 2)[0][0]
idx2 = np.where(arr1 == 3)[0][0]

# 交换元素
arr1[idx1], arr1[idx2] = arr1[idx2], arr1[idx1]

print(arr1)  # [1 3 2]

使用advanced indexing

在高级索引中，我们可以将要交换的值作为索引值传递给数组。但是这种方法只适用于一维数组。

import numpy as np

# 创建一个一维数组
arr1 = np.array([1, 2, 3])

# 交换元素
arr1[[1, 2]] = arr1[[2, 1]]

print(arr1)  # [1 3 2]

如果想要在二维数组中交换两个值，则需要对行和列进行分别的交换。

import numpy as np

# 创建一个二维数组
arr2 = np.array([[1, 2, 3], [4, 5, 6]])

# 交换第1行和第2行
arr2[[0, 1], :] = arr2[[1, 0], :]

# 交换第1列和第2列
arr2[:, [0, 1]] = arr2[:, [1, 0]]

print(arr2)
"""
[[5 4 6]
 [2 1 3]]
"""