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

解决思路:

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

环境搭建准备:

需要提前下载好numpy模块。

下载示范:

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

Numpy解决找出二维随机矩阵中每行数据中最接近某个数字的数字_第1张图片

 

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

Numpy解决找出二维随机矩阵中每行数据中最接近某个数字的数字_第2张图片

Numpy解决找出二维随机矩阵中每行数据中最接近某个数字的数字_第3张图片

 

     3.输入如下命令:

pip install numpy

Numpy解决找出二维随机矩阵中每行数据中最接近某个数字的数字_第4张图片

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


具体实现过程:

  • np.random.rand()函数生成随机的矩阵

    Numpy解决找出二维随机矩阵中每行数据中最接近某个数字的数字_第5张图片

 

a = np.random.rand(10,3)
  • abs绝对值函数

Numpy解决找出二维随机矩阵中每行数据中最接近某个数字的数字_第6张图片

 

b = abs(a-0.5)
  • np.argsort()函数对数组下标排序

Numpy解决找出二维随机矩阵中每行数据中最接近某个数字的数字_第7张图片

 

c = np.argsort(b)

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

  • mask提取指定行中的元素

Numpy解决找出二维随机矩阵中每行数据中最接近某个数字的数字_第8张图片

 

mask = c[:,0]
  • for循环输出

Numpy解决找出二维随机矩阵中每行数据中最接近某个数字的数字_第9张图片

 

for i in range(10):
    print(a[i,mask[i]])

完整代码演示:

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]])

你可能感兴趣的:(python机器学习,数据分析,numpy,矩阵,python)