列取决于csv数据的随机化

时间:2012-08-09 09:27:43

标签: python

我正在使用一个名为psychopy的程序,它是用python编写的,大部分时间它都有一个GUI就足够了,但你可以输入你自己的代码段。基本上我遇到的问题是我有一个csv文件,我得到python导入,在csv文件中有六列和一些代码我可以得到python随机化每列中的顺序:

import random 
import csv 

stimulilist=list(csv.reader(open('original_file.csv',"rU")))

stimuliones=[]
locationones=[]
locationtwos=[]
stimulitwos=[]
locationthrees[]
locationfours[]

for i in range(len(stimulilist)):
    stimuliones.append(stimulilist[i][0])
    locationones.append(stimulilist[i][1])
    locationtwos.append(stimulilist[i][2])
    stimulitwos.append(stimulilist[i][3])
    locationthrees.append(stimulilist[i][4])
    locationfours.append(stimulilist[i][5])

Stim1=stimuliones[1:]
location1=locationones[1:]
location2=locationtwos[1:]
Stim2=stimuliones[1:]
location3=locationthrees[1:]
location4=positionfours[1:]

然后我将它们随机化,这将为第一批试验创建我的两个随机单词列表。

Stimuli1Random=random.sample(Stim1,len(Stim1)) 
Location1Random=random.sample(location1,len(location1))
Location2Random=random.sample(location2,len(location2))
Stimuli2Random=random.sample(Stim2,len(Stim2))
Location3random=random.sample(location3,len(location3))
Location4random=random.sample(location4,len(location4))

然而,我希望刺激1,位置1和位置2一起随机化,刺激2位置3和位置4一起随机化,例如:

CSV文件中的内容:

(row)    Stim1 Location1 Location2         (row)     Stim2 Location3 Location4
1        P1    Left      Left               1        c1    Left      Left
2        p2    Middle    Middle             2        c2    Middle    Middle
3        p3    Right     Right              3        c3    Right     Right
4        p4    Left      Left               4        c4    Left      Left
5        p5    Middle    Middle             5        c5    Middle    Middle
6        p6    Right     Right              6        c6    Right     Right

使用上面的代码,每个列都可以单独随机化,例如:

(row)    Stim1 Location1 Location2         (row)   Stim2  Location3 Location4
1        P6    Left      Middle             1      c5     Right     Left
2        p1    Right     Middle             2      c3     Middle    Middle
3        p5    Middle    Right              3      c1     Left      Middle
4        p4    Left      Left               4      c2     Right     Left
5        p3    Middle    Left               5      c4     Middle    Right
6        p2    Right     Right              6      c6     Left      Right

但我想让行随机化,以便它看起来像这样:

(row)    Stim1 Location1 Location2       (row)  Stim2  Location3 Location4
1        p2    Middle    Middle           1     c3      Right   Right
2        p6    Right     Right            2     c5      Middle  Middle
3        p3    Right     Right            3     c4      Left    Left
4        p4    Left      Left             4     c1      Left    Left
5        p5    Middle    Middle           5     c2      Middle  Middle
6        P1    Left      Left             6     c6      Right   Right

所以基本上我希望Stim1,Location 1和Location 2的行保持不变但是随机化,好像它们是同一列,但是,我不想将它们合并到一列,重要的是它们保持个性列。我也想要这个用于刺激位置3和位置4.

另外,假设我想从csv文件中导入另一列,但不能随机化它,我该怎么把它放到代码中?

1 个答案:

答案 0 :(得分:0)

如果您想将这些数据部分随机化,最简单的方法是将它们存储在一起。您甚至不必为此更改csv文件的结构。如果将单个部分存储在一个列表中(而不是单独存储在每个列中),则可以随机化该列表,相关部分将保持在一起:

stimulilist=list(csv.reader(open('original_file.csv',"rU")))

stim_1_data = []
stim_2_data = []

for i in range(len(stimulilist)):
    stim_1_data.append(stimulilist[i][0:3])
    stim_2_data.append(stimulilist[i][3:6])

random.shuffle(stim_1_data)
random.shuffle(stim_2_data)

如果您现在想要获取stim1和stim2的数据,您可以从每个列表中取出一个元素,并根据需要分离这些元素:

stim1, locaion1, location2 = stim_1_data.pop()
stim2, locaion3, location4 = stim_1_data.pop()

如果您想要csv文件中的其他列,只需添加另一个不随机化的列表:

stimulilist=list(csv.reader(open('original_file.csv',"rU")))

stim_1_data = []
stim_2_data = []
non_random_data = []

for i in range(len(stimulilist)):
    stim_1_data.append(stimulilist[i][0:3])
    stim_2_data.append(stimulilist[i][3:6])
    non_random_data.append(stimulilist[i][6])

random.shuffle(stim_1_data)
random.shuffle(stim_2_data)

如果您需要列表,就像您最初使用每个值分开的那样,您可以再次提取它们:

Stim1, location1,location2 = zip(*stim_1_data)
Stim2, location3,location4 = zip(*stim_2_data)