随机播放 Python 中的集合字典

Question

我正在尝试随机排列集合字典。这是我要随机洗牌的组

{'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 的新手。谢谢你的帮助。

Answer 1

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

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']}

随机播放 Python 中的集合字典

Shuffle dictionaries of sets in Python

python

random

dictionary

shuffle

set