zoukankan      html  css  js  c++  java
  • python 寻找可迭代(list)的目标元素的下表方法

    python中sorted()排序后返回同时返回下标

    nums = [4, 1, 5, 2, 9, 6, 8, 7]
    sorted_nums = sorted(enumerate(nums), key=lambda x: x[1])
    idx = [i[0] for i in sorted_nums]
    nums = [i[1] for i in sorted_nums]

    除了使用 emuerate()函数外,还可以使用zip()

    nums = [4, 1, 5, 2, 9, 6, 8, 7]
    idenx = list(range(len(nums)))
    sorted_nums = zip(idenx,nums)
    sorted_nums = sorted(enumerate(nums), key=lambda x: x[1])
    idx = [i[0] for i in sorted_nums]
    nums = [i[1] for i in sorted_nums]

    两者本质是一样的

    另外一种方法是将list转化为numpy,这样就可以使用numpy的内置函数np.where()

     注意np.where返回的数据格式,返回的是2个array数组,一个存放的是1维坐标,另一个存放的二维坐标,这里的A是二维,如果n维会生成n个array。

     或者是np.argmax()  函数 

     

      

  • 相关阅读:
    Fizz Buzz 问题
    旋转字符串
    合并排序数组
    尾部的零
    A + B 问题
    CentOS6.x安装RabbitMQ
    MySql游标
    MySql存储过程
    找出n个自然数(1,2,3,……,n)中取r个数的组合
    正则表达式
  • 原文地址:https://www.cnblogs.com/cuc-lyp/p/13797911.html
Copyright © 2011-2022 走看看