3. 读取CSV文件并显示前几行

读取CSV文件并显示前几行 学习卡片

任务说明

本案例要求使用Pandas读取一个CSV文件,并展示其前几行内容。我们将创建一个示例CSV文件(包含销售记录),然后通过pd.read_csv()加载数据,并使用head()方法显示前5行(默认值),同时验证数据读取的正确性。

实现代码

import pandas as pd

# 1. 创建示例CSV文件(模拟销售数据)
csv_content = """order_id,product,quantity,price,order_date
1001,笔记本电脑,2,5999,2024-03-01
1002,无线鼠标,5,129,2024-03-02
1003,机械键盘,3,399,2024-03-03
1004,显示器,1,1899,2024-03-04
1005,USB集线器,4,89,2024-03-05
1006,摄像头,2,299,2024-03-06
1007,耳机,3,499,2024-03-07
"""

with open("sales_data.csv", "w", encoding="utf-8") as f:
    f.write(csv_content)

# 2. 使用Pandas读取CSV文件
df = pd.read_csv("sales_data.csv")

# 3. 显示前几行(默认前5行)
print("=== 前5行数据 ===")
print(df.head())

# 4. 额外验证:显示前3行
print("\n=== 前3行数据 ===")
print(df.head(3))

# 5. 输出数据基本信息以确认读取成功
print("\n=== 数据形状 ===")
print(f"共 {df.shape[0]} 行, {df.shape[1]} 列")
print("\n=== 列名 ===")
print(list(df.columns))

代码说明

代码首先在本地创建一个包含7条销售记录的CSV文件sales_data.csv,字段包括订单号、产品名、数量、单价和日期。随后使用pd.read_csv()读取该文件为DataFrame对象。通过head()方法展示前5行数据,并额外展示前3行以演示参数用法。最后输出数据形状和列名,用于确认文件被正确解析。

运行与验证

运行方式:在Python环境中执行上述代码(需已安装Pandas,可通过pip install pandas安装)。代码会在当前目录生成sales_data.csv文件并读取。

预期输出:控制台将依次显示前5行数据(包含表头)、前3行数据、数据形状(7行5列)以及列名列表。前5行输出应如下:

   order_id     product  quantity  price  order_date
0      1001    笔记本电脑         2   5999  2024-03-01
1      1002      无线鼠标         5    129  2024-03-02
2      1003      机械键盘         3    399  2024-03-03
3      1004        显示器         1   1899  2024-03-04
4      1005    USB集线器         4     89  2024-03-05

判定标准:若输出结果与上述格式一致,且df.shape显示为(7, 5),列名包含order_id、product、quantity、price、order_date,则说明CSV读取与head()显示功能正确实现。

CSV读取与head()显示流程 sales_data.csv order_id, product, quantity, price, order_date (7条记录) pd.read_csv() DataFrame 7行 × 5列 索引: 0~6 列: 5个字段 .head() 前5行数据 默认 head() 显示行: 0~4 含表头行 验证观测点 head() 默认返回前5行,包含列名表头 head(3) 可指定行数,返回前3行 df.shape 输出 (7, 5) 确认完整读取