Python将CSV文件数据读取至数组或矩阵的两种方法

花开有时
花开有时
发布时间:2026-10-13| 6人浏览过
原创

本文详解利用Numpy和Pandas库将CSV文件数据读取为数组或矩阵的方法。重点解决路径转义字符导致的SyntaxError,并演示使用Pandas处理缺失值及结合Sklearn的Imputer进行数据清洗的完整流程。

首先介绍如何使用Numpy库来实现这一功能。

需要注意的是,该方法存在一个明显缺点:如果CSV文件中包含缺失值,读取过程将会失败。

读取数据的代码如下:

import numpy 
my_matrix = numpy.loadtxt(open("1.csv","rb"),delimiter=",",skiprows=0) 

将矩阵数据写入CSV文件的代码如下:

numpy.savetxt('2.csv', my_matrix, delimiter = ',')

在操作过程中可能会遇到以下错误:

SyntaxError: (unicode error) 'unicodeescape' codec can't decode bytes in position

出现该问题的原因是文件路径字符串中包含了转义字符,解决方法是将路径中的反斜杠\替换为双反斜杠\\。

接下来介绍利用Pandas库进行读取的方法。

读取数据示例(Pandas可以自动处理缺失值):

>>> import pandas as pd
>>> df = pd.read_csv('D:\Python\\l\B_train1.csv')
>>> df.values
>>> df.as_matrix(columns=None)

写入数据示例(直接将DataFrame对象保存为CSV):

>>> testB = test[test.intersection_id.isin(["B"])]
>>> pd.DataFrame.to_csv(testB,"D:\Python\\k\\t5B.csv")
#testB是个dataFrame

若需利用Sklearn包中的Imputer模块来处理缺失值,可参考以下代码:

>>> m = df.as_matrix(columns=None)
>>> from sklearn.preprocessing import Imputer
>>> imp = Imputer(missing_values='NaN', strategy='mean', axis=0)
>>> imp.fit(m)
Imputer(axis=0, copy=True, missing_values='NaN', strategy='mean', verbose=0)
>>> imp.transform(m)

以上就是Python读取CSV文件数据到数组(矩阵)的完整实例讲解,希望能为读者提供有效的参考。

免责声明:本文内容仅供参考,不代表本站观点。转载或引用请注明出处。

Python 相关产品和应用

全部下载
Selenium

Selenium

4.51.0 · Windows

用于浏览器自动化和 Web 测试的 Python 工具。

scikit-learn

scikit-learn

1.9.1 · Windows

常用 Python 机器学习算法与模型评估工具库。

Black

Black

26.10.1 · Windows

强调一致性的 Python 自动代码格式化工具。

Ruff

Ruff

0.17.1 · macOS

由 Rust 编写的高速 Python 代码检查和格式化工具。

pytest

pytest

9.1.1 · Windows

简洁而强大的 Python 自动化测试框架。

FastAPI

FastAPI

0.146.0 · Windows

高性能 Python API 框架,支持类型提示和自动接口文档。

Flask

Flask

3.1.3 · Windows

轻量灵活的 Python Web 应用框架。

Django

Django

6.1.2 · Windows

成熟的 Python Web 框架,适合构建内容站点和业务系统。

Spyder

Spyder

6.1.7 · Windows

面向科学计算与数据分析的 Python 集成开发环境。

SciPy

SciPy

1.18.1 · Windows

面向科学、工程和技术计算的 Python 算法库。