pandasDataFrame读和写csv文件

2024-01-09 04:31:24

从.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)

文章来源:https://blog.csdn.net/weixin_44194001/article/details/135407485
本文来自互联网用户投稿，该文观点仅代表作者本人，不代表本站立场。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。