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

最新推荐文章于 2024-06-10 17:15:00 发布

BUNNYGG

最新推荐文章于 2024-06-10 17:15:00 发布

阅读量499

点赞数

CC 4.0 BY-SA版权

文章标签： pycharm python ide

本文链接：https://blog.csdn.net/BUNNYGG/article/details/121532711

本文介绍了如何使用Python读取和写入CSV文件，以及如何处理txt文件的读写，包括数据分割和标签整理。重点在于文件操作技巧与基础数据预处理步骤。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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()