时间戳

首页 > 解决方案 > 在 Python 中随机播放集合字典

python - 在 Python 中随机播放集合字典

问题描述

我正在尝试对集合的字典进行洗牌。这是我想随机洗牌的集合

{'1': ['IMG_0001.png', 'IMG_0002.png', 'IMG_0003.png', 'IMG_0004.png'], 
 '2': ['IMG_0020.png', 'IMG_0021.png', 'IMG_0022.png', 'IMG_0023.png'], 
 '3': ['IMG_0051.png', 'IMG_0052.png', 'IMG_0053.png', 'IMG_0054.png']}

输出应该有点像下面这样

{'1': ['IMG_0001.png', 'IMG_0002.png', 'IMG_0053.png', 'IMG_0054.png'], 
 '2': ['IMG_0020.png', 'IMG_0021.png', 'IMG_0022.png', 'IMG_0023.png'], 
 '3': ['IMG_0003.png', 'IMG_0004.png', 'IMG_0051.png', 'IMG_0052.png']}

尝试在 python 中使用 random.shuffle() 但没有达到预期的效果。谁能帮我。我是python的新手。谢谢您的帮助。

标签： pythondictionaryrandomsetshuffle

解决方案

如果每个列表应该具有相同的大小，并且您希望将列表中的所有元素相互打乱，您可以使用以下函数：

import random
import itertools


def shuffle(original: dict) -> dict:
    all_elements = list(itertools.chain.from_iterable(original.values()))
    m = len(all_elements) // len(original)  # list size
    random.shuffle(all_elements)
    return {k: all_elements[i*m : (i+1)*m] for i, k in enumerate(original)}

或者，使用more-itertools包：

import random
import itertools
import more_itertools


def shuffle(original: dict) -> dict:
    all_elements = list(itertools.chain.from_iterable(original.values()))
    list_length = len(all_elements) // len(original)
    random.shuffle(all_elements)
    return {k: l for k, l in more_itertools.sliced(all_elements, n=list_length)}

示例用法：

d = {
    "1": ["IMG_0001.png", "IMG_0002.png", "IMG_0003.png", "IMG_0004.png"],
    "2": ["IMG_0020.png", "IMG_0021.png", "IMG_0022.png", "IMG_0023.png"],
    "3": ["IMG_0051.png", "IMG_0052.png", "IMG_0053.png", "IMG_0054.png"],
}
print(shuffle(d))

{'1': ['IMG_0023.png', 'IMG_0003.png', 'IMG_0021.png', 'IMG_0001.png'],
 '2': ['IMG_0053.png', 'IMG_0051.png', 'IMG_0052.png', 'IMG_0004.png'],
 '3': ['IMG_0054.png', 'IMG_0022.png', 'IMG_0002.png', 'IMG_0020.png']}

推荐阅读