python read_csv读取CSV文件|极客教程

python read_csv读取CSV文件

CSV（逗号分隔值）是一种常见的文件格式，用来存储表格数据，类似于Excel表格。Python中有很多方法可以读取和处理CSV文件，本文将介绍如何使用Python读取CSV文件并进行简单处理。

读取CSV文件

Python中有很多库可以用来读取CSV文件，比如csv模块、pandas库等。我们将分别介绍这两种方法。

使用csv模块

csv模块是Python内置的模块，用于处理CSV文件。通过csv.reader可以逐行读取CSV文件中的数据。

import csv

# 打开CSV文件
with open('data.csv', 'r') as file:
    reader = csv.reader(file)

    # 读取每一行数据
    for row in reader:
        print(row)

上面的代码打开名为data.csv的CSV文件，并逐行读取数据，然后打印出来。假设data.csv文件内容如下：

Name,Age,Gender
Alice,25,Female
Bob,30,Male

运行上面的代码会输出：

['Name', 'Age', 'Gender']
['Alice', '25', 'Female']
['Bob', '30', 'Male']

可以看到，csv.reader以列表的形式返回每一行的数据。

使用pandas库

pandas是一个强大的数据处理库，可以方便地读取和处理CSV文件。通过pandas.read_csv函数可以将CSV文件读取为DataFrame对象。

import pandas as pd

# 读取CSV文件为DataFrame
df = pd.read_csv('data.csv')

# 打印DataFrame
print(df)

假设data.csv文件内容如上，运行上面的代码会输出：

    Name  Age  Gender
0  Alice   25  Female
1    Bob   30    Male

可以看到，pandas.read_csv函数将CSV文件读取为DataFrame对象，更方便地进行数据处理。

处理CSV文件

读取CSV文件之后，我们可以对数据进行一些处理，比如筛选特定的行或列，统计数据等。

筛选特定的行或列

通过csv模块或pandas库，我们可以筛选出特定的行或列。

# 使用csv模块筛选特定列
with open('data.csv', 'r') as file:
    reader = csv.reader(file)

    # 只读取Name列
    for row in reader:
        print(row[0])

# 使用pandas库筛选特定列
print(df['Name'])

上面的代码分别使用csv模块和pandas库来筛选出Name列的数据。

统计数据

通过pandas库，我们可以很方便地对数据进行统计分析。

# 统计年龄的平均值
print(df['Age'].mean())

# 统计每个性别的人数
print(df['Gender'].value_counts())

上面的代码分别统计了年龄的平均值和每个性别的人数。

总结

本文介绍了如何使用Python读取和处理CSV文件。通过csv模块和pandas库，我们可以很方便地对CSV文件中的数据进行操作。

python read_csv读取CSV文件