使用CSV模块读取CSV文件--含代码

简介: 使用CSV模块读取CSV文件--含代码

使用CSV某块读取CSV文件

` CSV文件开头有多行说明,有汉字,有效数据是ASCII,直接读取时数据无法被代码使用

导入模块:

import os # 用于遍历目录中的文件。
import numpy as np # 用于处理 numpy 矩阵(虽然在这个示例中主要用 pandas)。
import matplotlib.pyplot as plt # 用于画图,本文未使用

读取单个CSV文件

定义 read_csv_to_numpy_matrix 函数:

  • file_path 是单个 CSV 文件的完整路径。
  • start_row 从指定行开始读取。

初始化列表:

data = [] 用于存储单个 CSV 文件的数据矩阵。

打开文件、关闭文件

with open(file_path, mode='r', newline='', encoding='utf-8') as file:
file.close()

读取文件

csv_reader = csv.reader(file)

跳过标题行

header = next(csv_reader)

读取指定行的内容

for current_row, row in enumerate(csv_reader, start=1):
    if current_row >= start_row: # 从指定行开始读取数据
        data.append([float(x) for x in row])

将数据转换为 numpy 矩阵

matrix = np.array(data)

返回数据

return matrix

读取目录下所有CSV文件

定义 read_csv_files_to_dict 函数:

  • directory_path 是包含 CSV 文件的目录路径。
  • start_row 从指定行开始读取。

初始化字典:

matrices = {} 用于存储每个 CSV 文件的数据矩阵。

遍历目录中的文件:

使用 os.listdir(directory_path) 获取目录中的所有文件。

检查文件名是否以 .csv 结尾。

读取和处理 CSV 文件:

使用 read_csv_to_numpy_matrix(file_path, start_row) 读取 CSV 文件到 DataFrame。

存储矩阵到字典:

将矩阵放入 matrices 字典中,键为文件名。

返回字典:

函数返回包含所有 CSV 文件数据矩阵的字典。

示例用法:

调用 read_csv_files_to_dict 函数,并将结果存储在 csv_matrices 变量中。

遍历字典并打印每个文件名及其对应的 numpy 矩阵。

注意事项

字典中的键是文件名(包括扩展名)。如果你希望不包括扩展名,可以在存储到字典时进行相应处理。

完整代码

python读取csv文件及读取多个文件,将字符串转成numpy

相关文章
ChatGPT高效提问—prompt基础
ChatGPT高效提问—prompt基础
391 0
|
数据可视化 数据处理 数据库
【Python篇】PyQt5 超详细教程——由入门到精通(中篇一)
【Python篇】PyQt5 超详细教程——由入门到精通(中篇一)
1084 2
|
11月前
|
机器学习/深度学习 人工智能 编译器
【AI系统】微分实现方式
本文详细介绍了自动微分的三种实现方法:基本表达式、操作符重载和源代码转换。每种方法都有其特点和适用场景,包括它们的实现原理、优缺点。自动微分是机器学习和深度学习中的关键技术,理解这些实现方式有助于更好地掌握其背后的数学原理和工程实践。文中还提到了具体的应用案例和工具,如PyTorch和MindSpore,展示了这些方法在实际项目中的应用。
248 3
|
12月前
|
JavaScript 前端开发 开发者
探索 DrissionPage: 强大的Python网页自动化工具
DrissionPage 是一个基于 Python 的网页自动化工具,结合了浏览器自动化的便利性和 requests 库的高效率。它提供三种页面对象:ChromiumPage、WebPage 和 SessionPage,分别适用于不同的使用场景,帮助开发者高效完成网页自动化任务。
1001 4
|
消息中间件 监控 Ubuntu
大数据-54 Kafka 安装配置 环境变量配置 启动服务 Ubuntu配置 ZooKeeper
大数据-54 Kafka 安装配置 环境变量配置 启动服务 Ubuntu配置 ZooKeeper
386 3
大数据-54 Kafka 安装配置 环境变量配置 启动服务 Ubuntu配置 ZooKeeper
|
JavaScript API
NodeJs——使用axios下载上传文件
NodeJs——使用axios下载上传文件
384 4
|
敏捷开发 监控 Java
阿里云云效产品使用合集之Codeup WebIDE环境下,如何使用通义灵码
云效作为一款全面覆盖研发全生命周期管理的云端效能平台,致力于帮助企业实现高效协同、敏捷研发和持续交付。本合集收集整理了用户在使用云效过程中遇到的常见问题,问题涉及项目创建与管理、需求规划与迭代、代码托管与版本控制、自动化测试、持续集成与发布等方面。
|
自然语言处理 数据可视化 搜索推荐
用Python制作酷炫词云图,原来这么简单!
用Python制作酷炫词云图,原来这么简单!
2151 1
|
数据采集
爬虫案例—抓取找歌词网站的按歌词找歌名数据
爬虫案例—抓取找歌词网站的按歌词找歌名数据
296 0