Gist to copy data from csv to .txt

gistfile1.txt

"""
It read files from csv file and separates "labels" and "texts" from it.
"""
import pandas as pd
import os
import numpy as np
import string


def run():
    """
    run to separate rows
    :return:
    """
    file_sep_dir = os.getcwd()
    msg_label_dir = os.getcwd() + '/msg_labels'
    msg_label_file = msg_label_dir + '/msg_labels_file.txt'

    msg_texts_dir = os.getcwd() + '/msg_texts'
    msg_texts_file = msg_texts_dir + '/msg_texts_file.txt'

    if not os.path.isfile(file_sep_dir + "/temp_spam_data.csv"):
        print('---file not found!')
        return None

    data = pd.read_csv('temp_spam_data.csv', sep=",")
    labels = data[['label']]
    texts = data[['text']]

    # turning them to 0. & 1.
    labels = [1. if label == 'spam' else 0. for label in labels.values]
    # saving labels
    if not os.path.isdir(msg_label_dir):
        os.mkdir(msg_label_dir)
    np.savetxt(msg_label_file, labels, fmt='%f')
    print('---saving labels done!')

    # saving texts
    texts = [np.array_str(text) for text in texts.values]
    texts = [x.lower() for x in texts]
    texts = [''.join(c for c in x if c not in string.punctuation) for x in texts]
    texts = [''.join(c for c in x if c not in '0123456789') for x in texts]
    texts = [' '.join(x.split()) for x in texts]

    if not os.path.isdir(msg_texts_dir):
        os.mkdir(msg_texts_dir)
    np.savetxt(msg_texts_file, texts, delimiter=' ', fmt='%s')
    print('---saving texts done!')
    print('---all done!')


if __name__ == '__main__':
    run()

msg_labels_file.txt

0.000000
0.000000
1.000000
0.000000

msg_texts_file.txt

go until jurong point crazy available only in bugis n great world la e buffet cine there got amore wat
ok lar joking wif u oni
free entry in a wkly comp to win fa cup final tkts st may text fa to to receive entry questionstd txt ratetcs apply overs
u dun say so early hor u c already then say

temp_spam_data.csv