python - 同时翻转 Numpy 数组中的两个值

标签 python numpy

如果我有一个 numpy 数组,例如:

[0,1,0,2,2]

我想同时翻转列表中的 0 和 2(以获得 [2,1,2,0,0]),最好的方法是什么?

最佳答案

这是条件在 numpy 中的直接应用。

def switchvals(arr, val1, val2):
    mask1 = arr == val1
    mask2 = arr == val2
    arr[mask1] = val2
    arr[mask2] = val1

关于python - 同时翻转 Numpy 数组中的两个值,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/31757047/

相关文章:

python - Performantly argsort 两个预排序的 numpy 数组

python - 是否有替代算法在 Python 中执行以下任务(天文管道)

python - 如何将 .mat 二进制格式转换为 2D numpy 数组?

python - 使用 NumPy 将用户/项目评分转换为二维数组

python - Python 中多个数组的最大值和最小值

python - 有没有办法让这段代码更通用、更优雅?

python - 全路径算法的优化

python - 如何每小时从 Linux 终端运行 python 脚本?

python - k-means 集群中的项数

python - Coverage.py 中的 "No source for code"消息