Numpy对二维矩阵中每一行的最大元素乘以2 其余不变

Posted 2022-11-22 海轰Pro

tags:

篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了Numpy对二维矩阵中每一行的最大元素乘以2 其余不变相关的知识，希望对你有一定的参考价值。

简介

Hello！
非常感谢您阅读海轰的文章，倘若文中有错误的地方，欢迎您指出～

ଘ(੭ˊᵕˋ)੭
昵称：海轰
标签：程序猿｜C++选手｜学生
简介：因C语言结识编程，随后转入计算机专业，获得过国家奖学金，有幸在竞赛中拿过一些国奖、省奖…已保研
学习经验：扎实基础 + 多做笔记 + 多敲代码 + 多思考 + 学好英语！

唯有努力💪

本文仅记录自己感兴趣的内容

场景描述

存在一个二维矩阵（ndarry类型），需要进行以下操作

找到每一行的最大元素，将其乘以2
其余元素不变

解决方法

import numpy as np

a = np.array([[10, 12],
                                   [10, 3],
                                   [3, 4]])
print(a)

a = np.where(a == [[i] for i in np.max(a, axis=1)], a * 2, a)

print(a)

举一反三

如果是对每一列的最大元素乘以2，其余元素不变呢

很容易想到一个方法：先转置，再求每一行最大值，最后再转置一次即可

这里其实可以这样

import numpy as np

a = np.array([[10, 12],
                                   [10, 3],
                                   [3, 4]])

print(a)

a = np.where(a == [i for i in np.max(a, axis=0)], a * 2, a)
print(a)

这里与之前的不同有两个

axis = 0（这里比较好理解，找列最大值）
i for i in np.max() 而不是之前的 [i] for i in np.max()（这里有点难理解）

测试一下

print([i for i in np.max(a, axis=0)])

print([[i] for i in np.max(a, axis=1)])

这里应该就是在使用a == [i for i in np.max(a, axis=0)] 时，自动对后半部分进行了填充！

比如a == [i for i in np.max(a, axis=0)]

虽然 [i for i in np.max(a, axis=0)] = [10, 12]

但是填充成与a维度一样，其余元素为0

[i for i in np.max(a, axis=0)] = [[10,12],[0,0],[0,0]]（这里只是以0简单举个例子，不是真的填充为0）

import numpy as np

a = np.array([[10, 12],
                                   [0, 0],
                                   [0, 0]])

print(a == [i for i in np.max(a, axis=0)])