import numpy as np

def computeEquationNumber(nDof,element_data):
    nElements=len(element_data)
    numEquations=[]
    for i in range(nElements):
        elementConnectivity=element_data[i][1]
        if elementConnectivity[0]==elementConnectivity[1]: #spring element
            numEquationsElem=[elementConnectivity[0]*nDof+element_data[i][2]]
        else:
            numEquationsElem=[]
            for j in range(len(elementConnectivity)):
                for k in range(nDof):
                    numEquationsElem.append(elementConnectivity[j]*nDof+k)
        numEquations.append(numEquationsElem)
    
    return numEquations