保存多个随机播放的CSV文件

时间:2019-11-25 13:18:28

标签: python list csv random shuffle

有没有一种快速方法可以将列表随机播放n次(以不同的顺序)并将其保存为n个单个csv文件?我已经搜索了很多,但是却找不到任何东西。我有以下代码,但是我敢肯定它会更短,因此我不确定所有改组后的列表的顺序都不同。有解决方案吗?

import random

example = ['S01_a', 'S01_b', 'S02_a', 'S02_b', 'S03_a', 'S03_b', 'S04_a']

while True:
    example

    shuffle3 = []

    last = ""

    while example:
        i = example
        if last:
            i = [x for x in i if x[4] != last[4]]
        if not i:
            #no valid solution
            break 
        newEl = random.choice(i)
        last = newEl
        shuffle3.append(newEl)
        example.remove(newEl)
    if not example:
        break


fid = open("example.csv", encoding='latin1', mode="w")
fid.writelines(shuffle3)
fid.close()

3 个答案:

答案 0 :(得分:1)

您可以在列表索引上生成所有可能的排列,然后按照排列给定的顺序选择元素以生成新的随机排列的列表。最后,随机播放列表并选择第一个N。

from itertools import permutations
from random import shuffle


example = ['S01_a', 'S01_b', 'S02_a', 'S02_b', 'S03_a', 'S03_b', 'S04_a']
indices = [x for x in range(0,len(example))]
n_perm = 5

all_permutations = list(set(permutations(indices)))
shuffle(all_permutations)
my_permutations = all_permutations[:n_perm]       

for index, elem in enumerate(my_permutations):

    new_shuffle = [example[x] for x in elem]    
    with open("example_{}.csv".format(str(index)), "w") as fid:
        fid.writelines(",".join(new_shuffle))

答案 1 :(得分:0)

您可以这样做:

import itertools, random
N = 5
example = ['S01_a', 'S01_b', 'S02_a', 'S02_b', 'S03_a', 'S03_b', 'S04_a']
all_options = list(itertools.permutations(example, len(example)))
my_lists = random.choices(all_options, k=N)
my_lists

输出:

[('S02_b', 'S01_a', 'S03_a', 'S01_b', 'S02_a', 'S03_b', 'S04_a'),
 ('S03_b', 'S02_b', 'S01_b', 'S03_a', 'S01_a', 'S02_a', 'S04_a'),
 ('S02_a', 'S04_a', 'S03_a', 'S02_b', 'S03_b', 'S01_b', 'S01_a'),
 ('S02_b', 'S04_a', 'S01_a', 'S02_a', 'S03_b', 'S01_b', 'S03_a'),
 ('S03_a', 'S04_a', 'S01_b', 'S02_a', 'S02_b', 'S01_a', 'S03_b')]

然后,如果您想分别对它们做某事,只需循环它们:

for l in my_lists:
    I_do_what_it_want with l...

答案 2 :(得分:-2)

像这样的东西?您必须寻找一个库来生成csv

import java.util.*; 

public class GFG 
{ 
    public static void main(String[] args) 
    { 
        ArrayList<String>  mylist = new ArrayList<String>(); 
        mylist.add("code"); 
        mylist.add("quiz"); 
        mylist.add("geeksforgeeks"); 
        mylist.add("quiz"); 
        mylist.add("practice"); 
        mylist.add("qa"); 

        System.out.println("Original List : \n" + mylist); 

        // Here we use Random() to shuffle given list. 
        Collections.shuffle(mylist, new Random()); 
        System.out.println("\nShuffled List with Random() : \n"
                                                     + mylist); 

        // Here we use Random(3) to shuffle given list. 
        Collections.shuffle(mylist, new Random(3)); 
        System.out.println("\nShuffled List with Random(3) : \n"
                                                     + mylist); 

        // Here we use Random(3) to shuffle given list. 
        Collections.shuffle(mylist, new Random(5)); 
        System.out.println("\nShuffled List with Random(5) : \n"
                                                     + mylist); 

    } 
}