Python Numpy 实现交换两行和两列的方法

林见深
林见深
发布时间:2026-10-13| 4人浏览过
原创

本文详解在Python Numpy中交换数组行与列的两种方法。针对直接赋值导致的引用错误,推荐使用np.copy()或花式索引a[[1,2], :] = a[[2,1], :]。同时演示了通过列索引重排实现列交换,确保数据操作准确无误。

Numpy数组行与列交换方法概述

Numpy是Python中广泛使用的科学计算库。尽管网络上关于交换数组行的资料较多,但专门讲解如何高效交换两列的内容相对较少。经过查阅官方文档及相关资源,整理出以下实用的交换方法。

交换两行的实现与陷阱

交换两行

假设有一个Numpy数组a = np.array([[1,2,3],[2,3,4],[1,6,5], [9,3,4]]),现在需要交换第二行和第三行的数据。初学者可能会尝试直接进行赋值操作,代码如下:


import numpy as np

a = np.array([[1,2,3],[2,3,4],[1,6,5], [9,3,4]])
tmp = a[1]
a[1] = a[2]
a[2] = tmp

执行上述代码后,输出结果如下:


array([[1, 2, 3],
  [1, 6, 5],
  [1, 6, 5],
  [9, 3, 4]])

出现该错误的原因在于,语句tmp = a[1]并未创建a[1]的副本,而是生成了一个指向同一内存区域的“别名”。因此,后续对a[1]的修改会同步影响tmp。修正后的代码应显式创建副本:


import numpy as np

a = np.array([[1,2,3],[2,3,4],[1,6,5], [9,3,4]])
tmp = np.copy(a[1])
a[1] = a[2]
a[2] = tmp

此时,运行结果符合预期:


array([[1, 2, 3],
  [1, 6, 5],
  [2, 3, 4],
  [9, 3, 4]])

虽然上述方法可行,但是否存在更简洁的写法?查阅文档后发现,利用Numpy的高级索引功能可以实现更优雅的交换方式:


import numpy as np

a = np.array([[1,2,3],[2,3,4],[1,6,5], [9,3,4]])
a[[1,2], :] = a[[2,1], :]

交换两列的操作技巧

交换两列

交换列的逻辑与交换行类似,主要通过调整列索引顺序来实现。具体操作如下:


a = np.array([[1,2,3],[2,3,4],[1,6,5], [9,3,4]])
a[:,[1,0,2]]
a

最终得到的数组状态为:


array([[1, 2, 3],
  [2, 3, 4],
  [1, 6, 5],
  [9, 3, 4]])

以上内容即为Python Numpy中交换两行和两列的具体实现方法,供各位开发者参考。如有其他疑问或建议,欢迎交流探讨。

相关标签:

免责声明:本文内容仅供参考,不代表本站观点。转载或引用请注明出处。

Python 相关产品和应用

全部下载
Selenium

Selenium

4.51.0 · Windows

用于浏览器自动化和 Web 测试的 Python 工具。

scikit-learn

scikit-learn

1.9.1 · Windows

常用 Python 机器学习算法与模型评估工具库。

Black

Black

26.10.1 · Windows

强调一致性的 Python 自动代码格式化工具。

Ruff

Ruff

0.17.1 · macOS

由 Rust 编写的高速 Python 代码检查和格式化工具。

pytest

pytest

9.1.1 · Windows

简洁而强大的 Python 自动化测试框架。

FastAPI

FastAPI

0.146.0 · Windows

高性能 Python API 框架,支持类型提示和自动接口文档。

Flask

Flask

3.1.3 · Windows

轻量灵活的 Python Web 应用框架。

Django

Django

6.1.2 · Windows

成熟的 Python Web 框架,适合构建内容站点和业务系统。

Spyder

Spyder

6.1.7 · Windows

面向科学计算与数据分析的 Python 集成开发环境。

SciPy

SciPy

1.18.1 · Windows

面向科学、工程和技术计算的 Python 算法库。