Pandas读取csv文件某一列并保存到txt文件中

Pandas读取csv文件某一列并保存到txt文件中

按道理来说挺简单的，但是却查了好久，本来想利用pandas的api获取某一列的数据，然后写入到文件中，但是写入到文件后出现各种问题，要不程序报错，要不不分行。最后还是按照之前的那样，获取的之后一个个的添加到之前创建好的list中。

def save_csv_to_text(filename, csv_name, usecols):
    '''
    读取csv的文件，将指定列转换存到txt文件中，usecols=0 摘要，usecols=1 文章
    '''
    data = pd.read_csv(os.path.join(DATA_ROOT, csv_name), usecols=[usecols])
    data_list = data.values.tolist()
    result = []
    for item in data_list:
        result.append(item[0])
    print("start process {}".format(filename))
    start_time = time.time()
    with open(filename, 'w', encoding='utf-8') as f:
        for item in tqdm(result):
            f.write(item + '\n')
    f.close()
    print("cost time {}".format(time.time() - start_time))
    print('save {} done!'.format(filename))
    print("---------------------")

在这里tqdm如果想使用的话，导包的时候需要这样导入：

from tqdm import tqdm