unsupervised.py 文件源码

def makeTable(df, rowsCol, colsCol, dataCol):
    # df.set_index(rowsCol)

    uniqRowVals = pd.unique(df[rowsCol])
    uniqColVals = pd.unique(df[colsCol])

    # "rows col = ", df[rowsCol]
    # print "uniq row vals", uniqRowVals
    # print "uniq col vals", uniqColVals
    # print df[[rowsCol, colsCol, dataCol]]

    out = pd.DataFrame(index=uniqRowVals, columns=uniqColVals)
    for rowVal in uniqRowVals:
        for colVal in uniqColVals:
            rowsMatch = df[rowsCol] == rowVal
            colsMatch = df[colsCol] == colVal
            thisIdx = np.where(rowsMatch * colsMatch)[0][0]
            out.ix[rowVal][colVal] = df[dataCol][thisIdx]

    return out