这是一个
NBack test
.
当我按原样执行代码时,它就工作了。屏幕上显示的顺序是
self.sequence = np.array([0,0,0,1,1,1,6,6,6,0,1,2,3,4,5,6,7,8,0,1])
但是我有一个函数,它可以生成我想要使用的序列,而不是那个数组。是ZBack.py
当我叫我的Zback.py时
self.sequence = generateZBackSequence(20, 5)
程序不工作,而不是数组。
它输出此错误:
Exception in thread Thread-14:
Traceback (most recent call last):
File "C:\ProgramData\Anaconda3\lib\threading.py", line 917, in _bootstrap_inner
self.run()
File "C:\ProgramData\Anaconda3\lib\threading.py", line 865, in run
self._target(*self._args, **self._kwargs)
File "C:/Users/Python_Scripts/N-back.py", line 198, in targetTask
for i in range(self.sequence.shape[0]):
AttributeError: 'NBack' object has no attribute 'sequence'
我做错什么了?为什么它告诉我属性“sequence”不存在?
我已经检查了压痕。我的zback.py也会返回一个数组。
import tkinter as tk
from tkinter import Tk
from threading import Thread
from keylogger.KeyLogger import KeyLogger
from time import sleep
from nBackTools.NBackTools import *
from nBackTools.ZBack import *
from random import randint
import numpy as np
import csv
def _create_circle(self, x, y, r, **kwargs):
return self.create_oval(x-r, y-r, x+r, y+r, **kwargs)
tk.Canvas.create_circle = _create_circle
class NBack:
def __init__(self, master):
##Title of the window
self.master = master
master.title("N-Back")
##It measures the screen size (width x height + x + y)
##The opened window will be based on the screen size
master.geometry("{0}x{1}-0+0".format(master.winfo_screenwidth(), master.winfo_screenheight()))
self.canvas = tk.Canvas(master, width=master.winfo_screenwidth(), height=master.winfo_screenheight(), \
borderwidth=0, highlightthickness=0, bg="grey")
self.canvasWidth = master.winfo_screenwidth()
self.canvasHeight = master.winfo_screenheight()
self.createLines()
self.createText()
self.canvas.grid()
self.t = Thread(target=self.targetTask,)
self.t.daemon = True
self.t.start()
## Positions of the circles ("stims")
## - -
## 0 - 1 - 2
## ------------
## 3 - 4 - 5
## ------------
## 6 - 7 - 8
## - -
self.positions = np.array([[(self.canvasWidth/2)-130, (self.canvasHeight/2)-130],\
[(self.canvasWidth/2), (self.canvasHeight/2)-130],\
[(self.canvasWidth/2)+130, (self.canvasHeight/2)-130],\
[(self.canvasWidth/2)-130, (self.canvasHeight/2)],\
[(self.canvasWidth/2), (self.canvasHeight/2)],\
[(self.canvasWidth/2)+130, (self.canvasHeight/2)],\
[(self.canvasWidth/2)-130, (self.canvasHeight/2)+130], \
[(self.canvasWidth/2), (self.canvasHeight/2)+130], \
[(self.canvasWidth/2)+130, (self.canvasHeight/2)+130]])
###############################################
###############################################
#THIS IS THE PROBLEM!!!
#self.sequence = generateZBackSequence(20, 5)
self.sequence = np.array([0,0,0,1,1,1,6,6,6,0,1,2,3,4,5,6,7,8,0,1])
###############################################
###############################################
"""
ADDING LINES TO THE SCREEN
"""
def createLines(self, linewidth = 5):
#Vertical 1
self.canvas.create_line((self.canvasWidth/2)-65, (self.canvasHeight/2)-65-130, \
(self.canvasWidth/2)-65, (self.canvasHeight/2)+65+130, width= linewidth )
#Vertical 2
self.canvas.create_line((self.canvasWidth/2)+65, (self.canvasHeight/2)-65-130, \
(self.canvasWidth/2)+65, (self.canvasHeight/2)+65+130, width= linewidth )
#Horizontal1
self.canvas.create_line((self.canvasWidth/2)-65-130, (self.canvasHeight/2)-65, \
(self.canvasWidth/2)+65+130, (self.canvasHeight/2)-65, width= linewidth )
#Horizontal2
self.canvas.create_line((self.canvasWidth/2)-65-130, (self.canvasHeight/2)+65, \
(self.canvasWidth/2)+65+130, (self.canvasHeight/2)+65, width= linewidth )
"""
ADDING TEXT TO THE SCREEN
"""
def createText(self):
self.canvas.create_text((self.canvasWidth/4), (self.canvasHeight - self.canvasHeight/4), text="A=YES", font=( None, 20))
self.canvas.create_text((self.canvasWidth/4), (self.canvasHeight - self.canvasHeight/4)+50, text="S=NO", font=( None, 20))
"""
CIRCLES APPEAR AND DISAPPEAR
"""
def targetTask(self):
thisnb = randint(75,900*2)
targetFile = open(str(thisnb)+"_targets.csv",'w', newline='')
writer = csv.writer(targetFile)
##Loops from 0 to the Number of circles that will appear (default=20)
for i in range(self.sequence.shape[0]):
posX = self.positions[self.sequence[i]][0]
posY = self.positions[self.sequence[i]][1]
print(": (%d,%d)" % (posX,posY))
##Records the coordenates in an excel file
writer.writerow([posX,posY])
## How many seconds each circle will appear
circle = self.canvas.create_circle(posX, posY, 55, fill="black", outline="#000", width=1)
sleep(1)
## Delete circle
## How many seconds circle dissapears
self.canvas.delete(circle)
sleep(2)
sleep(1)
print('All done')
targetFile.close()
self.master.destroy()
root = Tk()
my_gui = NBack(root)
root.mainloop()
这是zback.py:
import numpy as np
import random
def generateZBackSequence(nbTrial, nbOfYes):
##all possible positions of the "stims"
allStims = [0,1,2,3,4,5,6,7,8]
##sequence is an array of -1s and has the size of nbTrial (default=20)
sequence = np.zeros((nbTrial,))-1
##security: in case the required "yeses" are too many
if nbOfYes > nbTrial:
print('generateZBackSequence: invalid nb of Yeses')
return None
##Pick randomly 1 stim and repeat it (default=5)
yeses = np.repeat(random.sample(allStims, 1),nbOfYes)
print("yeses:", yeses, '\n')
##Making the first value in the sequence = value in yeses
sequence[0]=yeses[0]
#print("sequence:", sequence, '\n')
##Contains the "stims" that were not randomly sampled
newStims = [x for x in allStims if x not in yeses]
#print("newStims:", newStims, '\n')
"""
STEP 1 - impose the yeses
"""
##nbOfYes-1, because there is already a "yeses" in sequence[0]
for i in range(nbOfYes-1):
idxValid = False
while idxValid==False:
##select index randomly, from 1 to 19
idx = random.randint(1, nbTrial-1)
if sequence[idx]==-1:
idxValid = True
print("idx:", idx, '\n')
##insert yeses in the random index
sequence[idx] = yeses[i]
#print("sequence_idx:", sequence)
"""
STEP 2 - fill in the sequence
"""
##Makes a "for" from 0 to 19
for i in range(sequence.shape[0]):
if sequence[i]==-1:
stimValid = False
##Search for -1s
while stimValid==False:
##Takes a random sample "stim" which is different from 'yeses'
stimID = random.sample(newStims, 1)
##stimID is a list of 1 element, sequence is an array of 20 elements
sequence[i]=stimID[0]
stimValid = True
else:
stimValid = True
return sequence
def _test():
print(generateZBackSequence(20,5))
print("")
print(np.arange(0,20,1))
pass
if __name__ == "__main__":
_test()
