python将数字放入数组,让它们尽量均匀的分布

写这个是因为一个热心的豆瓣网友给我出了个思考题。菜菜如我,思考了一下,给了一份代码。

def distribution(list):
    import numpy as np
    #生成的列表
    new_list = []

    #统计有哪些数
    set_list = np.unique(list)

    #统计每个数出现的次数
    count_list = []
    for i in set_list:
        count_list.append(list.count(i))

    #生成字典
    list1 = dict(zip(set_list,count_list))
    print('list1',list1)

    #出现次数排序
    from operator import itemgetter
    list2 =[index for index,value in sorted(enumerate(list1),key=itemgetter(1),reverse=True)]
    print('list2',list2)

    count_number = 0
    for i in list2:
        if list1[i] == 1:#出现次数为1的数字
            new_list.append(i)
            count_number += 1
        elif list1[i] != 1:
            k = list1[i]
            count_number += k
            for j in range(k):
                new_list.insert(j * int(count_number/k),i)

    print(new_list)
    return new_list

list = [0,0,0,0,1,1,1,0,2,3,0,0,0,1,2,4]
distribution(list)

结果。

list1 {0: 8, 1: 4, 2: 2, 3: 1, 4: 1}
list2 [4, 3, 2, 1, 0]
[0, 1, 0, 2, 0, 1, 0, 4, 0, 1, 0, 2, 0, 1, 0, 3]

好像还可以吧。

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值