pandasDataFrame读和写csv文件

从.csv文件读数据

import pandas as pd# 从CSV文件中读取数据
train_df = pd.read_csv("datasets/train01.csv")
val_df = pd.read_csv("datasets/val01.csv")
test_df = pd.read_csv("datasets/test01.csv")# 显示数据框的前几行，确保数据已成功加载
print("Train Data:")
print(train_df.head())print("\nValidation Data:")
print(val_df.head())print("\nTest Data:")
print(test_df.head())

把数据写到.csv文件

# 假设你有一个名为output_df的DataFrame
output_df = pd.DataFrame({'Column1': [1, 2, 3],'Column2': ['A', 'B', 'C']
})# 将DataFrame写入CSV文件
output_df.to_csv("output_file.csv", index=False)

train_df[["label", "cutword"]].to_csv("datasets/train01.csv", index=False)
val_df[["label", "cutword"]].to_csv("datasets/val01.csv", index=False)
test_df[["label", "cutword"]].to_csv("datasets/test01.csv", index=False)