3. 读取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()显示功能正确实现。