Python 读取与写入 csv/txt 操作(自用)

csv

读取

import pandas
df = pandas.read_csv('data.csv') ## 可加参数 sep = ';', header, index_col, names = ['col_A','col_B']
print(df)

写入

test_x_df = pd.DataFrame(real_test_x, columns=['contents'])
test_y_df = pd.DataFrame(real_test_y, columns=['labels'])
test_set = pd.concat([test_x_df, test_y_df], axis=1)

test_set['data'].to_csv('test_set.csv',index = False, header = False)

Txt

读取

contents = []
labels = []
with open(path, mode="r", encoding="UTF-18") as f:
    for line in tqdm(f):
        line = line.strip()
        content, label = line.split("\t")
        contents.append(content)
        labels.append(int(label))

写入

all_labels = list(set(train_set['labels']))

with open('label.txt', 'w') as f:
    for i in range(len(all_labels)):
        line = all_labels[i] + '\n'
        f.write(line)
    f.close()