python获取列表中最大的N个数索引

针对数组无重复数的

import heapq
int n
lis=[2,4,5,1,7]
re1 = map(lis.index, heapq.nlargest(n, lis)) #求最大的n个索引    nsmallest 求最小  nlargest求最大
re2 = heapq.nlargest(n, lis) #求最大的三个元素
print(list(re1)) #因为re1由map()生成的不是list,直接print不出来,添加list()就行了
print(re2) 

有重复数


import heapq
lis= [2, 4, 4, 1, 0]
int n
max_number = heapq.nlargest(n, lis) 
max_index = []
for t in max_number:
    index = lis.index(t)
    max_index.append(index)
    lis[index] = 0
    
print(max_number)
print(max_index)
  • 5
    点赞
  • 12
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
### 回答1: 从n个数选出m个数的算法可以使用递归的方法进行实现。首先,我们需要定义一个函来选择: ```python def choose_numbers(arr, m, selected_numbers=[]): if m == 0: return [selected_numbers] result = [] for i in range(len(arr)): num = arr[i] remaining_numbers = arr[i+1:] result += choose_numbers(remaining_numbers, m-1, selected_numbers+[num]) return result ``` 在这个函,`arr`参是待选择的n个数列表,`m`参表示要选择的个数,`selected_numbers`参是已经选择的列表。 接下来,我们通过调用这个函来选择: ```python n = int(input("请输入n的值:")) m = int(input("请输入m的值:")) arr = [] for i in range(n): arr.append(int(input("请输入第{}个数:".format(i+1)))) selected_numbers = choose_numbers(arr, m) for num_list in selected_numbers: print(num_list) ``` 在这段代码,我们首先通过`input`函获取用户输入的n和m的值。然后,我们通过循环获取用户输入的n个数,并将其添加到`arr`列表。最后,我们调用`choose_numbers`函来选择,并将结果打印输出。 这个算法会输出所有可能的选择结果。例如,当n=4,m=2时,选择的结果为:[(1, 2), (1, 3), (1, 4), (2, 3), (2, 4), (3, 4)]。 注意:由于题目并没有明确要求是否可以重复选择,所以上述算法允许重复选择同一个数。如果不允许重复选择同一个数,可以在选择时进行判断并排除重复选择的情况。 ### 回答2: 从n个数选m个数的算法可以使用递归的方式来实现。以下是一个用Python编写的递归算法实现示例: ```python def combination(nums, m): result = [] current = [] def helper(start, nums, m): if m == 0: # 如果已选满m个数,则将结果保存到结果列表 result.append(current[:]) return if start >= len(nums): # 如果起始位置超过了组长度,则返回 return current.append(nums[start]) # 选择当前 helper(start + 1, nums, m - 1) # 递归调用,继续选择下一个数 current.pop() # 回溯,撤销选择 helper(start + 1, nums, m) # 不选择当前,继续向后遍历 helper(0, nums, m) return result ``` 使用示例: ```python nums = [1, 2, 3, 4, 5] m = 3 result = combination(nums, m) print(result) # 输出 [[1, 2, 3], [1, 2, 4], [1, 2, 5], [1, 3, 4], [1, 3, 5], [1, 4, 5], [2, 3, 4], [2, 3, 5], [2, 4, 5], [3, 4, 5]] ``` 以上算法通过递归实现了从n个数选取m个数的所有组合情况,并将结果保存在result列表返回。 ### 回答3: 要从n个数选m个数,可以使用回溯法实现。下面是一个用Python编写的示例代码: ```python def combination(nums, m): result = [] # 存放选取的组合结果 path = [] # 存放当前选取的路径 def backtrack(nums, start): if len(path) == m: # 到达指定的选取个数m result.append(path[:]) return for i in range(start, len(nums)): path.append(nums[i]) backtrack(nums, i+1) # 递归进入下一层,注意是 i+1 path.pop() # 回溯,尝试下一个分支 backtrack(nums, 0) # 从索引0开始回溯 return result nums = [1, 2, 3, 4, 5] m = 3 print(combination(nums, m)) ``` 以上代码,`num`表示给定的n个数列表,`m`表示要选取的个数。`result`列表用于存放最终的组合结果,`path`列表用于存放当前选取的路径。 在`backtrack`函,首先判断当前选取的个数是否达到目标个数`m`,如果是则将当前路径加入结果列表,否则进行递归。 在每一层递归,从当前索引`start`开始遍历`nums`列表,将当前元素加入路径,并递归进入下一层。递归返回后,将当前元素从路径移除,进行回溯,继续尝试下一个分支。 最后调用`combination`函并输出结果,即可得到从n个数选m个数的所有组合。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值