前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >Numpy解决找出二维随机矩阵中每行数据中最接近某个数字的数字

Numpy解决找出二维随机矩阵中每行数据中最接近某个数字的数字

作者头像
用户10271432
发布2022-12-19 14:36:41
5290
发布2022-12-19 14:36:41
举报
文章被收录于专栏:机器学习-大数据

解决思路:

  1. 利用np.random.rand()函数生成随机的矩阵。
  2. abs函数实现对矩阵中每一个元素和指定元素相减
  3. np.argsort()函数实现找到排序后新元素在原来矩阵中的下标
  4. 利用mask函数提取矩阵中第一列的元素
  5. 最后利用for循环遍历所有的二维坐标,找到矩阵中每行中满足特定要求的数字

环境搭建准备:

需要提前下载好numpy模块。

下载示范:

  1. win+r进入运行窗口。

     2.输入cmd,进入命令行窗口

     3.输入如下命令:

代码语言:javascript
复制
pip install numpy

包安装好之后,就可以开始正常写代码了 


具体实现过程:

  • np.random.rand()函数生成随机的矩阵
代码语言:javascript
复制
a = np.random.rand(10,3)
  • abs绝对值函数
代码语言:javascript
复制
b = abs(a-0.5)
  • np.argsort()函数对数组下标排序
代码语言:javascript
复制
c = np.argsort(b)

注意到c数组中第一列的元素,表示的b中最小的元素在b中的下标,利用mask对其进行提取数据

  • mask提取指定行中的元素
代码语言:javascript
复制
mask = c[:,0]
  • for循环输出
代码语言:javascript
复制
for i in range(10):
    print(a[i,mask[i]])

完整代码演示:

代码语言:javascript
复制
a = np.random.rand(10,3)
b = abs(a-0.5)
c = np.argsort(b)
mask = c[:,0]
for i in range(10):
    print(a[i,mask[i]])
本文参与 腾讯云自媒体同步曝光计划,分享自作者个人站点/博客。
原始发表:2022-12-03,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • 解决思路:
  • 环境搭建准备:
  • 具体实现过程:
  • 完整代码演示:
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档