Shuffle two lists in the same order in python

Question

I have a question about shuffling, but first, here is my code:

from psychopy import visual, event, gui
import random, os
from random import shuffle
from PIL import Image
import glob
a = glob.glob("DDtest/targetimagelist1/*")
b = glob.glob("DDtest/distractorimagelist1/*")
target = a
distractor = b
pos1 = [-.05,-.05]
pos2 = [.05, .05]

shuffle(a)
shuffle(b)

def loop_function_bro():
    win = visual.Window(size=(1280, 800), fullscr=True, screen=0, monitor='testMonitor', color=[-1,-1,-1], colorSpace='rgb')
        distractorstim = visual.ImageStim(win=win,
        image= distractor[i], mask=None,
        ori=0, pos=pos1, size=[0.5,0.5],
        color=[1,1,1], colorSpace='rgb', opacity=1,
        flipHoriz=False, flipVert=False,
        texRes=128, interpolate=True, depth=-1.0)

    targetstim= visual.ImageStim(win=win,
        image= target[i], mask=None,
        ori=0, pos=pos2, size=[0.5,0.5],
        color=[1,1,1], colorSpace='rgb', opacity=1,
        flipHoriz=False, flipVert=False,
        texRes=128, interpolate=True, depth=-2.0)

    distractorstim.setAutoDraw(True)
    targetstim.setAutoDraw(True)
    win.flip()
    event.waitKeys(keyList = ['space'])

for i in range (2):  
    loop_function_bro()

This code randomly shuffles a bunch of images and displays them. However, I would like it is shuffle the images in the same order so that both lists are displayed in the same random order. Is there an way to do this?

Cheers, :)

Why not just shuffle the list of indices that you want to use (ie `range(len(image_list))`)? — , Feb 24 '15 at 18:24
possible duplicate of [How can I have changes made to one list affect another, different list?](http://stackoverflow.com/questions/28549985/how-can-i-have-changes-made-to-one-list-affect-another-different-list) — Andy, Feb 24 '15 at 18:33
Regardless of shuffling, in the code above you should not be re-creating the window on each loop iteration. Create it just once at the start of the experiment. Similarly, just create the two stimuli once. Then, in your loop_function_bro(), just do this: targetstim.setImage(a(indices[i]); distractorstim.setImage(b(indices[i])) #using the indices list as suggested in the answer by @Ken below. — Michael MacAskill, Feb 24 '15 at 20:33

score 6 · Answer 1 · edited Feb 24 '15 at 18:35

6

The simplest way I can think to handle this is use a separate list for indices, and shuffle it instead.

indices = list(xrange(len(a)))  # Or just range(len(a)) in Python 2
shuffle(indices)

edited Feb 24 '15 at 18:35

Kijewski

25,517
12
101
143

answered Feb 24 '15 at 18:25

Spice

352
1
5

Regarding your comment, if you use Python3, it should be `list(range(len(a)))`. – Evpok Feb 24 '15 at 18:30
That's true. I meant for it to work for both, but forgot `xrange` doesn't exist in Python 3. – Spice Feb 24 '15 at 18:32
I tried using this code, but I'm not quite sure how to implement it into the my program - sorry I'm new. :( – y3trgfhsfgr Feb 24 '15 at 19:11
1

range(len(a)) is sufficient, as it returns a list anyway. – Michael MacAskill Feb 24 '15 at 20:36
1

Not in Python 3, unfortunately. Using `range` in Python 3 (or `xrange`), you get an object which is an iterator. However, `shuffle` requires a mutable sequence (which is modified in-place). – Spice Feb 24 '15 at 20:41
Ken, do you know of any methods that work for Python 3? – y3trgfhsfgr Feb 24 '15 at 23:09
I erred a bit in my answer. For Python 3, you should use `list(range(len(a)))`. Or, for something that actually works in both 2 and 3, you can use `[i for i in range(len(a))]`. – Spice Feb 24 '15 at 23:17
Slow but brilliant. – JohnJohn Jan 26 '18 at 01:58

score 1 · Accepted Answer · edited May 23 '17 at 11:53

This question has been answered here and here. I particularly like the following for its syntactic simplicity

randomizedItems = map(items.__getitem__, indices)

For a complete working example, see code below. Note that I've changed quite a bit to make the code much shorter, cleaner and faster. See comments.

# Import stuff. This section was simplified.
from psychopy import visual, event, gui  # gui is not used in this script
import random, os, glob
from PIL import Image

pos1 = [-.05,-.05]
pos2 = [.05, .05]

# import two images sequences and randomize in the same order
target = glob.glob("DDtest/targetimagelist1/*")
distractor = glob.glob("DDtest/distractorimagelist1/*")
indices = random.sample(range(len(target)), len(target))  # because you're using python 2
target = map(target.__getitem__, indices)
distractor = map(distractor.__getitem__, indices)

# Create window and stimuli ONCE. Think of them as containers in which you can update the content on the fly.
win = visual.Window(size=(1280, 800), fullscr=True, screen=0, monitor='testMonitor', color=[-1,-1,-1])  # removed a default value
targetstim = visual.ImageStim(win=win, pos=pos2, size=[0.5,0.5])
targetstim.autoDraw = True
distractorstim = visual.ImageStim(win=win, pos=pos1, size=[0.5,0.5])  # removed all default values and initiated without an image. Set autodraw here since you didn't seem to change it during runtime. But feel free to do it where you please.
distractorstim.autoDraw = True

# No need to pack in a function. Just loop immediately. Rather than just showing two stimuli, I've set it to loop over all stimuli.
for i in range(len(target)):
    # set images. OBS: this may take some time. Probably between 20 and 500 ms depending mainly on image dimensions. Smaller is faster. It's still much faster than generating a full Window/ImageStim each loop though.
    distractorstim.image = distractor[i]
    targetstim.image = target[i]

    # Display and wait for answer
    win.flip()
    event.waitKeys(keyList = ['space'])

Another way to randomize the images while maintaining the pairing is something you'd probably do anyway sometime: put them together in a list of dictionaries, where each dict represents a trial. So rather than the two map lines, do:

trialList = [{'target':target[i], 'distractor':distractor[i]} for i in indices]

Try printing the trial list (print trialList) to see what it looks like. And then loop over trials:

for trial in trialList:
    # set images.
    targetstim.image = trial['target']
    distractorstim.image = trial['distractor']

    # Display and wait for answer. Let's record reaction times in the trial, just for fun.
    win.flip()
    response = event.waitKeys(keyList = ['space'], timeStamped=True)
    trial['rt'] = response[0][1]  # first answer, second element is rt.

When I run this code, I get "TypeError: _init_() got an unexpected keyword argument 'autoDraw' — y3trgfhsfgr, Feb 25 '15 at 16:05
Oh, I thought it could be set during init. Script is updated so that autoDraw is set just after initialization. — Jonas Lindeløv, Feb 25 '15 at 16:09
Wow, this is amazing - it works perfectly, thank you so much. You just made my week. :) — y3trgfhsfgr, Feb 25 '15 at 16:19

score 0 · Answer 3 · answered Feb 24 '15 at 18:25

0

I would create an array of numbers, shuffle it and sort the image lists both according to the shuffled numbers.

So both lists are shuffled in the same way.

answered Feb 24 '15 at 18:25

Alu

727
5
16

score 0 · Answer 4 · answered Feb 24 '15 at 18:26

0

    data1=[foo bar foo]
    data2=[bar foo bar]
    data3=[foo bar foo]
    alldata=zip((data1,data2,data3))
# now do your shuffling on the "outside" index of alldata then
    (data1shuff,data2shuff,data3shuff) = zip(*alldata)

answered Feb 24 '15 at 18:26

bob.sacamento

6,283
10
56
115

when I try this I get "ValueError: need more than 1 value to unpack". I think this happens because my a and b variables are using the glob function. – y3trgfhsfgr Feb 24 '15 at 23:08

Shuffle two lists in the same order in python

4 Answers4

Linked