将Python中的文本文件作为函数参数传递

Question

I am working on a data science beginner tutorial and just copied the code. But I am unable to get rid of the "invalid syntax" error in my function.

Already tried to change it into self but then I get an error when I try to "call" the function.

class DataAlgorithmus:
    def readDataSet("/Users/leo/Desktop/DS_Code/Example.txt"):

        fr = open("/Users/leonard/Desktop/DS_Code/Example.txt")  
        numberOfLines = len(fr.readlines())  
        returnMat = numpy.zeros((numberOfLines-1, 3))
        classLabelVector = []
        classColorVector = []
        fr = open("/Users/leonard/Desktop/DS_Code/Example.txt") 
        index = 0

        for line in fr.readlines(): 
            if index != 0:         
                line = line.strip()
                listFromLine = line.split('\t')
                returnMat[index-1, :] = listFromLine[1:4]

                classLabel = listFromLine[4]
                if classLabel == "Buero":
                    color = 'yellow'
                else:
                    color = 'blue'

                classLabelVector.append(classLabel)
                classColorVector.append(color)
            index += 1

        return returnMat, classLabelVector, classColorVector

    dataSet, classLabelVector, classColorVector = readDataSet("/Users/leonard/Desktop/DS_Code/Example.txt")

It says that this line: def readDataSet("/Users/leo/Desktop/DS_Code/Example.txt"):

has the following error: invalid syntax (<unknown>, line 3) pylint(syntax-error) [3,1]

Answer 1

Thanks, @jonrsharpe & @Iguananaut I edited my code and it works now!

def readDataSet(self, filename):

    fr = open(filename)  
    numberOfLines = len(fr.readlines())  
    returnMat = numpy.zeros((numberOfLines-1, 3))
    classLabelVector = []
    classColorVector = []
    fr = open(filename) 
    index = 0

    for line in fr.readlines(): 
    if index != 0:         
            line = line.strip()
            listFromLine = line.split('\t')
            returnMat[index-1, :] = listFromLine[1:4]

            classLabel = listFromLine[4]
            if classLabel == "Buero":
                color = 'yellow'
            else:
                color = 'blue'

            classLabelVector.append(classLabel)
            classColorVector.append(color)
        index += 1

    return returnMat, classLabelVector, classColorVector

dataSet, classLabelVector, classColorVector = readDataSet("/Users/leonard/Desktop/DS_Code/Example.txt")