extract_genre.py 文件源码

def exportGenre(df, genre):
    print ("Now exporting ", genre)
    pd.set_option('display.width', 120)

    df_genre = df[df['genre'] == genre]
    print(df_genre.shape)

    df_sample = df_genre.ix[np.random.choice(df_genre.index, 10000, replace=False)]
    #print(df_sample)
    print(df_sample.shape)

    with open("lyrics/" + genre + ".txt", "a") as f:
        for index, row in df_sample.iterrows():
                f.write("<S>\n" + row['lyrics'] + "\n<E>\n")