在Python中，常见的四种读取CSV文件的方式

阿罗的小小仓库

已于 2023-12-24 10:49:34 修改

阅读量5.1k

点赞数 18

分类专栏： python基础理解文章标签： python 开发语言

于 2023-12-23 21:07:32 首次发布

本文链接：https://blog.csdn.net/weixin_51659315/article/details/135174367

版权

16 篇文章 2 订阅

订阅专栏

本文比较了Python中使用内置csv模块、pandas库、numpy库以及基本文件操作读取CSV文件的方式，强调了pandas的DataFrame在数据分析方面的优势和numpygenfromtxt的底层功能。

摘要由CSDN通过智能技术生成

假设有一个名为 example.csv 的CSV文件，内容如下：

方式一、使用内置的‘csv’模块：

import csv

with open('example.csv', 'r') as file:
    reader = csv.reader(file)
    for row in reader:
        print(row)

输出为：

['Name', ' Age', ' City']
['John', ' 25', ' New York']
['Alice', ' 30', ' London']
['Bob', ' 22', ' Tokyo']

方式二、使用‘pandas’库：

import pandas as pd

df = pd.read_csv('example.csv')
print(df)

输出为：

  Name   Age       City
0  John    25   New York
1 Alice    30     London
2   Bob    22      Tokyo

方式三、使用‘numpy’库：

import numpy as np

data = np.genfromtxt('example.csv', delimiter=',', dtype='str')
print(data)

输出为：

[['Name' ' Age' ' City']
 ['John' ' 25' ' New York']
 ['Alice' ' 30' ' London']
 ['Bob' ' 22' ' Tokyo']]

方式四、使用‘open’和‘split’：

with open('example.csv', 'r') as file:
    lines = file.readlines()
    for line in lines:
        data = line.strip().split(',')
        print(data)

输出为：

['Name', ' Age', ' City']
['John', ' 25', ' New York']
['Alice', ' 30', ' London']
['Bob', ' 22', ' Tokyo']

总结：

Pandas的DataFrame提供了更丰富的数据结构和操作方法，特别适用于数据分析和处理。而numpy的‘genfromtxt’返回一个NumPy数组，csv模块提供了更底层的行-列数据。

关注

专栏目录