如何将DataFrame数据写入csv文件及读取

 更新时间:2024年09月18日 10:59:14   作者:geerduo  
在Python中进行数据处理时,经常会用到CSV文件的读写操作,当需要将list数据保存到CSV文件时,可以使用内置的csv模块,若data是一个list,saveData函数能够将list中每个元素存储在CSV文件的一行,但需要注意的是,默认情况下读取出的CSV数据类型为str

一、saveData函数

import csv
 
def saveData(filePath, data):
 
    with open(filePath, 'w+', newline='') as f:
        writer = csv.writer(f)
        for row in data:
            writer.writerow(row)

如果data是一个list,saveData函数会将list中的每个元素保存在csv文件中的一行。

但是存在一个问题:

csv文件中保存的数据如果直接读取出来是str类型,需要进行转换

二、将DataFrame写入csv文件

情况1

将list写入csv文件

init_configs = []
for i, cluster in enumerate(clusters):
    init_configs.append([f"cluster {i}:"])
    for _, row in cluster.iterrows():
        config = row.tolist()
        init_configs.append(config)
 
saveData('filePath', init_configs)

config = row.tolist()会将df中每一行数据都转变为list,然后append到init_configs中,通过调用saveData函数可以将每一行数据表示的list都写到csv文件中的一行。

这种情况生成的csv文件中每一行的各个元素之间都是用逗号隔开。

情况2

将嵌套list写入csv文件

init_configs = []
for i, cluster in enumerate(clusters):
    init_configs.append([f"cluster {i}:"])
    for _, row in cluster.iterrows():
        config = row.tolist()
        init_configs.append([config])
 
saveData('filePath', init_configs)

这种情况下init_configs中每个元素是一个二维list,由于CSV 文件是一种纯文本格式,它无法直接存储列表或其他复杂的数据结构。

当尝试将 [[1, 2, 3, 4, 5]] 这个二维嵌套list写入CSV文件时,它会被自动转换为字符串形式。

如下:

三、读取csv文件

情况1

如果按照上面的方法1写csv文件,可以按照如下方式将内容读出来:

import pandas as pd
 
with open(filePath, 'r') as f:
    lines = f.readlines()
 
filtered_lines = [line for line in lines if not line.startswith('cluster')]
 
configs = []
for line in filtered_lines:
    config = [int(x) for x in line.strip().split(',')]
    configs.append(config)
 
df = pd.DataFrame(configs)

可以用列表表达式的方式将csv文件中每一行的数据转为一个list。

情况2

如果将整个list当作字符串写到了csv文件中,可以按照如下方式将内容正确地读出来:

import pandas as pd
import ast
 
with open(filePath, 'r') as f:
    lines = f.readlines()
 
filtered_lines = [line for line in lines if not line.startswith('cluster')]
 
configs = []
for line in filtered_lines:
    config = ast.literal_eval(eval(line))
    configs.append(config)
 
df = pd.DataFrame(configs)

通过config = ast.literal_eval(eval(line))可以将变为字符串的list还原成原本的数据形式。

总结

以上为个人经验,希望能给大家一个参考,也希望大家多多支持脚本之家。

相关文章

  • Python实现自动化录制屏幕并生成视频

    Python实现自动化录制屏幕并生成视频

    这篇文章主要为大家详细介绍了如何在有另一个Python进程运行时,使用Python自动录制屏幕生成视频,并用以检查其执行情况,感兴趣的小伙伴可以了解下
    2026-03-03
  • 图文详解OpenCV中光流以及视频特征点追踪

    图文详解OpenCV中光流以及视频特征点追踪

    光流是空间运动物体在观察成像平面上的像素运动的瞬时速度,是利用图像序列中像素在时间域上的变化以及相邻帧之间的相关性来找到上一帧跟当前帧之间存在的相应关系,这篇文章主要给大家介绍了关于OpenCV中光流以及视频特征点追踪的相关资料,需要的朋友可以参考下
    2021-08-08
  • python GUI库图形界面开发之PyQt5布局控件QHBoxLayout详细使用方法与实例

    python GUI库图形界面开发之PyQt5布局控件QHBoxLayout详细使用方法与实例

    这篇文章主要介绍了python GUI库图形界面开发之PyQt5布局控件QHBoxLayout详细使用方法与实例,需要的朋友可以参考下
    2020-03-03
  • python Celery定时任务的示例

    python Celery定时任务的示例

    这篇文章主要介绍了python Celery定时任务的示例,小编觉得挺不错的,现在分享给大家,也给大家做个参考。一起跟随小编过来看看吧
    2018-03-03
  • 基于Python实现创建、修改或删除Excel命名区域

    基于Python实现创建、修改或删除Excel命名区域

    Excel命名区域是电子表格中一个重要的功能,它允许用户为单元格或单元格范围分配有意义的名称,本文将详细介绍如何使用Free Spire.XLS for Python 进行创建,编辑和删除命名区域等操作
    2025-10-10
  • Pytorch.nn.conv2d 过程验证方式(单,多通道卷积过程)

    Pytorch.nn.conv2d 过程验证方式(单,多通道卷积过程)

    今天小编就为大家分享一篇Pytorch.nn.conv2d 过程验证方式(单,多通道卷积过程),具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
    2020-01-01
  • Python中的面向对象编程详解(下)

    Python中的面向对象编程详解(下)

    这篇文章主要介绍了Python中的面向对象编程详解(下),本文讲解了继承、super关键字、重写、多重继承、类、实例和其他对象的内建函数、私有化等内容,需要的朋友可以参考下
    2015-04-04
  • 让Django支持Sql Server作后端数据库的方法

    让Django支持Sql Server作后端数据库的方法

    今天小编就为大家分享一篇让Django支持Sql Server作后端数据库的方法,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
    2018-05-05
  • python中把嵌套的列表合并成一个列表方法总结

    python中把嵌套的列表合并成一个列表方法总结

    python中list这种数据结构很常用到,下面这篇文章主要给大家介绍了关于python中把嵌套的列表合并成一个列表方法的相关资料,文中通过实例代码介绍的非常详细,需要的朋友可以参考下
    2022-11-11
  • pyinstaller参数介绍以及总结详解

    pyinstaller参数介绍以及总结详解

    这篇文章主要介绍了pyinstaller参数介绍以及总结详解,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧
    2019-07-07

最新评论