今天有个同事提了个需求:
穷尽给定13个的所有组合,从13个数中取5个,6个..。13个数的所有组合情况,并求出每种组合的中位值,如果取得个数是奇数,中位数就是中间那个数,如果是偶数,就取中间两个数的平均数。
实现如下:
#-*-coding:GBK -*-
#数组中位数函数
import numpy as np
# 待组合的列表
from itertools import combinations
#求组合的函数
def combine(temp_list, n):
'''根据n获得列表中的所有可能组合(n个元素为一组)'''
temp_list2 = []
for c in combinations(temp_list, n):
temp_list2.append(c)
return temp_list2
list1 = [1, 2, 3, 4, 5, 6,7,8,9,10,11,12,13]
end_list = []
#穷尽所有的组合
for i in range(5,len(list1)):
end_list.extend(combine(list1, i))
#输出组合到txt文件
with open('example.txt', 'w') as file:
for i in range(len(end_list)):
a=str(end_list[i])#一种组合
zws=str(np.median(end_list[i],axis=None))#求中位值
print(a+'====zsw===='+zws, file=file) #输出到txt文件
问题解决