引言
Pandas是Python中最流行的数据分析库之一,它提供了快速、灵活且富有表现力的数据结构,旨在使”关系”或”标签”数据的使用既简单又直观。本文将深入探讨如何高效使用Pandas进行数据分析,涵盖从基础操作到高级技巧的全方位指南。
1. Pandas基础数据结构
1.1 Series对象
Series是Pandas中最基本的数据结构之一,它是一维数组,能够保存任何数据类型(整数、字符串、浮点数、Python对象等)。
import pandas as pd
import numpy as np
# 创建一个简单的Series
s = pd.Series([1, 3, 5, np.nan, 6, 8])
print(s)
输出:
0 1.0
1 3.0
2 5.0
3 NaN
4 6.0
5 8.0
dtype: float64
1.2 DataFrame对象
DataFrame是一个二维表格型数据结构,可以看作是由多个Series组成的字典(共用相同的索引)。
# 创建DataFrame的多种方式
# 方法1:从字典创建
data = {
'Name': ['Alice', 'Bob', 'Charlie', 'David'],
'Age': [25, 30, 35, 40],
'City': ['New York', 'London', 'Paris', 'Tokyo']
}
df = pd.DataFrame(data)
print(df)
输出:
Name Age City
0 Alice 25 New York
1 Bob 30 London
2 Charlie 35 Paris
3 David 40 Tokyo
2. 数据读取与写入
Pandas支持多种格式的数据输入输出,包括CSV、Excel、JSON、SQL等。
2.1 读取CSV文件
# 读取CSV文件
df_csv = pd.read_csv('data.csv',
encoding='utf-8',
sep=',',
header=0,
index_col=0)
# 常用参数说明:
# encoding: 文件编码格式
# sep: 分隔符
# header: 指定作为列名的行号
# index_col: 将哪一列作为索引
2.2 读取Excel文件
# 读取Excel文件
df_excel = pd.read_excel('data.xlsx',
sheet_name='Sheet1',
header=0,
usecols='A:C')
# 常用参数说明:
# sheet_name: 指定工作表名称或索引
# usecols: 指定读取的列,可以是列名列表或'A:C'这样的范围
2.3 写入文件
# 写入CSV
df.to_csv('output.csv', index=False, encoding='utf-8')
# 写入Excel
df.to_excel('output.xlsx', index=False, sheet_name='Data')
3. 数据查看与选择
3.1 查看数据
# 查看前5行
print(df.head())
# 查看后5行
print(df.tail())
# 查看数据基本信息
print(df.info())
# 查看统计摘要
print(df.describe())
# 查看数据形状(行数,列数)
print(df.shape)
3.2 数据选择
# 选择单列
names = df['Name']
# 选择多列
subset = df[['Name', 'Age']]
# 通过标签选择行
row = df.loc[1] # 选择索引为1的行
# 通过位置选择行
row = df.iloc[1] # 选择第2行(从0开始)
# 通过标签选择多行
rows = df.loc[1:3] # 选择索引1到3的行(包含3)
# 通过位置选择多行
rows = df.iloc[1:3] # 选择第2到第3行(不包含第4行)
# 布尔索引
adults = df[df['Age'] > 25]
4. 数据清洗
数据清洗是数据分析中最重要的步骤之一,通常占整个分析过程的60-80%时间。
4.1 处理缺失值
# 检查缺失值
print(df.isnull().sum())
# 删除包含缺失值的行
df_cleaned = df.dropna()
# 删除包含缺失值的列
df_cleaned = df.dropna(axis=1)
# 填充缺失值
df_filled = df.fillna(0) # 用0填充
df_filled = df.fillna(method='ffill') # 用前一个值填充
df_filled = df.fillna(method='bfill') # 1用后一个值填充
# 填充不同列的不同值
df_filled = df.fillna({
'Age': df['Age'].mean(),
'City': 'Unknown'
})
4.2 处理重复值
# 检查重复行
print(df.duplicated().sum())
# 删除重复行
df_unique = df.drop_duplicates()
# 删除特定列的重复行
df_unique = df.drop_duplicates(subset=['Name'])
# 保留最后一个重复项
df_unique = df.drop_duplicates(subset=['Name'], keep='last')
4.3 数据类型转换
# 转换列的数据类型
df['Age'] = df['Age'].astype(int)
# 转换日期列
df['Date'] = pd.to_datetime(df['Date'])
# 转换为分类类型(节省内存)
df['Category'] = df['Category'].astype('category')
5. 数据转换与操作
5.1 添加/删除列
# 添加新列
df['Age_Group'] = df['Age'] // 10 * 10
# 基于条件创建列
df['Status'] = np.where(df['Age'] > 30, 'Senior', 'Junior')
# 删除列
df = df.drop('Age_Group', axis=1)
# 或者
del df['Age_Group']
5.2 apply函数
# 使用apply函数进行复杂转换
def calculate_age_group(age):
if age < 20:
return 'Teenager'
elif age < 35:
return 'Young Adult'
else:
return 'Adult'
df['Age_Group'] = df['Age'].apply(calculate_age_group)
# 使用lambda函数
df['Name_Length'] = df['Name'].apply(lambda x: len(x))
5.3 向量化操作
# 向量化操作比apply更快
# 好的做法
df['Age_In_10_Years'] = df['Age'] + 10
# 避免的做法(使用apply)
# df['Age_In_10_Years'] = df['Age'].apply(lambda x: x + 10)
6. 分组与聚合
6.1 groupby操作
# 基本分组
grouped = df.groupby('City')
# 多列分组
grouped = df.groupby(['City', 'Age_Group'])
# 聚合函数
agg_result = df.groupby('City')['Age'].agg(['mean', 'sum', 'count'])
# 多列不同聚合函数
agg_result = df.groupby('City').agg({
'Age': ['mean', 'max'],
'Name': 'count'
})
6.2 转换与过滤
# 转换(transform)
df['City_Mean_Age'] = df.groupby('City')['Age'].transform('mean')
# 过滤(filter)
filtered = df.groupby('City').filter(lambda x: x['Age'].mean() > 30)
7. 合并与连接
7.1 concat
# 垂直合并
df1 = pd.DataFrame({'A': [1, 2], 'B': [3, 4]})
df2 = pd.DataFrame({'A': [5, 6], 'B': [7, 8]})
result = pd.concat([df1, df2], axis=0) # 垂直合并
# 水平合并
result = pd.concat([df1, df2], axis=1) # 水平合并
7.2 merge
# 创建示例数据
left = pd.DataFrame({
'key': ['K0', 'K1', 'K2', 'K3'],
'A': ['A0', 'A1', 'A2', 'A3'],
'B': ['B0', 'B1', 'B2', 'B3']
})
right = pd.DataFrame({
'key': ['K0', 'K1', 'K2', 'K3'],
'C': ['C0', 'C1', 'C2', 'C3'],
'D': ['D0', 'D1', 'D2', 'D3']
})
# 内连接
result = pd.merge(left, right, on='key')
# 左连接
result = pd.merge(left, right, on='key', how='left')
# 右连接
result = pd.merge(left, right, on='key', how='right')
# 外连接
result = pd.merge(left, right, on='key', how='outer')
8. 时间序列处理
8.1 时间索引
# 创建时间索引
dates = pd.date_range('20230101', periods=6)
df_time = pd.DataFrame(np.random.randn(6, 4), index=dates, columns=list('ABCD'))
# 重采样
monthly = df_time.resample('M').mean() # 月度重采样
weekly = df_time.resample('W').sum() # 周度重采样
8.2 时间偏移
# 时间偏移
from pandas.tseries.offsets import DateOffset
df_time.index = df_time.index + DateOffset(months=1) # 所有日期加1个月
9. 性能优化技巧
9.1 使用适当的数据类型
# 优化内存使用
def optimize_memory(df):
start_mem = df.memory_usage().sum() / 1024**2
for col in df.columns:
col_type = df[col].dtype
if col_type != object:
c_min = df[col].min()
c_max = df[col].max()
if str(col_type)[:3] == 'int':
if c_min > np.iinfo(np.int8).min and c_max < np.iinfo(np.int8).max:
df[col] = df[col].astype(np.int8)
elif c_min > np.iinfo(np.int16).min and c_max < npinfo(np.int16).max:
df[col] = df[col].astype(np.int16)
elif c_min > np.iinfo(np.int32).min and c_max < np.iinfo(np.int32).max:
df[col] = df[col].astype(np.int32)
elif c_min > np.iinfo(np.int64).min and c_max < np.iinfo(np.int64).max:
df[col] = df[col].astype(np.int64)
else:
if c_min > np.finfo(np.float32).min and c_max < np.finfo(np.float32).max:
df[col] = df[col].astype(np.float32)
else:
df[col] = df[col].astype(np.float64)
else:
df[col] = df[col].astype('category')
end_mem = df.memory_usage().sum() / 1024**2
print(f'Memory usage reduced from {start_mem:.2f} MB to {end_mem:.2f} MB')
return df
9.2 避免链式索引
# 避免链式索引(可能返回副本而不是视图)
# 不推荐
# df[df['Age'] > 25]['Status'] = 'Senior'
# 推荐
df.loc[df['Age'] > 25, 'Status'] = 'Senior'
9.3 使用query方法
# 使用query进行快速筛选(对于大数据集更高效)
result = df.query('Age > 25 and City == "London"')
10. 高级技巧
10.1 窗口函数
# 滚动平均
df['Rolling_Mean'] = df['Age'].rolling(window=2).mean()
# 扩展窗口
df['Cumulative_Mean'] = df['Age'].expanding().mean()
# 指数加权移动平均
df['EWMA'] = df['Age'].ewm(span=2).mean()
10.2 透视表
# 创建透视表
pivot = pd.pivot_table(df,
values='Age',
index='City',
columns='Age_Group',
aggfunc='mean',
fill_value=0)
10.3 交叉表
# 创建交叉表
crosstab = pd.crosstab(df['City'], df['Age_Group'])
11. 实战案例:完整数据分析流程
让我们通过一个完整的案例来展示如何将这些技巧应用到实际工作中。
import pandas as pd
import numpy as np
# 1. 创建示例数据集
np.random.seed(42)
data = {
'CustomerID': range(1, 1001),
'Age': np.random.randint(18, 70, 1000),
'City': np.random.choice(['Beijing', 'Shanghai', 'Guangzhou', 'Shenzhen'], 1000),
'Purchase_Amount': np.random.uniform(10, 1000, 1000).round(2),
'Purchase_Date': pd.date_range('2023-01-01', periods=1000, freq='H'),
'Category': np.random.choice(['Electronics', 'Clothing', 'Food', 'Books'], 1000)
}
df = pd.DataFrame(data)
# 2. 数据清洗
# 添加一些缺失值
df.loc[0:50, 'Age'] = np.nan
df.loc[100:150, 'City'] = np.nan
# 处理缺失值
df['Age'].fillna(df['Age'].mean(), inplace=True)
df['City'].fillna('Unknown', inplace=True)
# 3. 特征工程
# 创建年龄分组
df['Age_Group'] = pd.cut(df['Age'],
bins=[0, 25, 35, 45, 100],
labels=['Young', 'Adult', 'Middle', 'Senior'])
# 创建购买金额分组
df['Purchase_Level'] = pd.cut(df['Purchase_Amount'],
bins=[0, 100, 500, 1000],
labels=['Low', 'Medium', 'High'])
# 4. 数据分析
# 按城市和年龄组统计
analysis = df.groupby(['City', 'Age_Group']).agg({
'CustomerID': 'count',
'Purchase_Amount': ['mean', 'sum', 'std']
}).round(2)
# 5. 高级分析
# 计算每个客户的总购买金额
customer_summary = df.groupby('CustomerID').agg({
'Purchase_Amount': 'sum',
'Purchase_Date': ['min', 'max'],
'City': 'first'
}).round(2)
# 计算购买频率
customer_summary['Purchase_Frequency'] = (
customer_summary[('Purchase_Date', 'max')] -
customer_summary[('Purchase_Date', 'min')]
).dt.days + 1
# 6. 结果导出
analysis.to_csv('city_age_analysis.csv')
customer_summary.to_excel('customer_summary.xlsx')
print("分析完成!")
print(f"数据集大小: {df.shape}")
print(f"处理后的数据:\n{df.head()}")
12. 常见问题与解决方案
12.1 SettingWithCopyWarning
# 问题:链式索引可能导致警告
# 不推荐
# df[df['Age'] > 25]['Status'] = 'Senior'
# 解决方案1:使用loc
df.loc[df['Age'] > 25, 'Status'] = 'Senior'
# 解决方案2:使用copy()
df_filtered = df[df['Age'] > 25].copy()
df_filtered['Status'] = 'Senior'
12.2 内存不足
# 分块读取大文件
chunk_size = 10000
chunks = pd.read_csv('large_file.csv', chunksize=chunk_size)
for chunk in chunks:
# 处理每个块
processed_chunk = chunk[chunk['value'] > 100]
# 保存或进一步处理
12.3 性能瓶颈
# 使用numba加速(对于数值计算)
from numba import jit
@jit(nopython=True)
def calculate_distance(x, y):
return np.sqrt(x**2 + y**2)
# 使用向量化代替循环
# 不推荐
# for i in range(len(df)):
# df.loc[i, 'distance'] = calculate_distance(df.loc[i, 'x'], df.loc[i, 'y'])
# 推荐
df['distance'] = calculate_distance(df['x'].values, df['y'].values)
13. 最佳实践总结
- 始终使用向量化操作:避免使用循环和apply,除非必要
- 优化数据类型:使用适当的数据类型可以大幅减少内存使用
- 使用loc/iloc:明确指定选择方式,避免链式索引
- 链式操作:合理使用链式操作可以使代码更简洁
- 分块处理大数据:对于超大数据集,使用chunksize参数
- 及时释放内存:处理完大数据后,使用del和gc.collect()释放内存
- 使用query方法:对于复杂筛选条件,query方法更高效
- 利用groupby的灵活性:agg、transform、filter各有用途
- 保持索引一致性:在合并操作前确保索引正确
- 文档化代码:为复杂的数据处理步骤添加注释
14. 总结
Pandas是一个功能强大的数据分析工具,掌握其高效使用方法可以显著提升数据分析的效率。本文从基础到高级,详细介绍了Pandas的核心功能和最佳实践。记住,高效使用Pandas的关键在于:
- 理解数据结构
- 掌握向量化操作
- 合理使用内存
- 熟练运用分组和聚合
- 避免常见陷阱
通过不断练习和实践,你将能够熟练运用Pandas解决各种数据分析问题。# 如何在Python中高效使用Pandas进行数据分析
引言
Pandas是Python中最流行的数据分析库之一,它提供了快速、灵活且富有表现力的数据结构,旨在使”关系”或”标签”数据的使用既简单又直观。本文将深入探讨如何高效使用Pandas进行数据分析,涵盖从基础操作到高级技巧的全方位指南。
1. Pandas基础数据结构
1.1 Series对象
Series是Pandas中最基本的数据结构之一,它是一维数组,能够保存任何数据类型(整数、字符串、浮点数、Python对象等)。
import pandas as pd
import numpy as np
# 创建一个简单的Series
s = pd.Series([1, 3, 5, np.nan, 6, 8])
print(s)
输出:
0 1.0
1 3.0
2 5.0
3 NaN
4 6.0
5 8.0
dtype: float64
1.2 DataFrame对象
DataFrame是一个二维表格型数据结构,可以看作是由多个Series组成的字典(共用相同的索引)。
# 创建DataFrame的多种方式
# 方法1:从字典创建
data = {
'Name': ['Alice', 'Bob', 'Charlie', 'David'],
'Age': [25, 30, 35, 40],
'City': ['New York', 'London', 'Paris', 'Tokyo']
}
df = pd.DataFrame(data)
print(df)
输出:
Name Age City
0 Alice 25 New York
1 Bob 30 London
2 Charlie 35 Paris
3 David 40 Tokyo
2. 数据读取与写入
Pandas支持多种格式的数据输入输出,包括CSV、Excel、JSON、SQL等。
2.1 读取CSV文件
# 读取CSV文件
df_csv = pd.read_csv('data.csv',
encoding='utf-8',
sep=',',
header=0,
index_col=0)
# 常用参数说明:
# encoding: 文件编码格式
# sep: 分隔符
# header: 指定作为列名的行号
# index_col: 将哪一列作为索引
2.2 读取Excel文件
# 读取Excel文件
df_excel = pd.read_excel('data.xlsx',
sheet_name='Sheet1',
header=0,
usecols='A:C')
# 常用参数说明:
# sheet_name: 指定工作表名称或索引
# usecols: 指定读取的列,可以是列名列表或'A:C'这样的范围
2.3 写入文件
# 写入CSV
df.to_csv('output.csv', index=False, encoding='utf-8')
# 写入Excel
df.to_excel('output.xlsx', index=False, sheet_name='Data')
3. 数据查看与选择
3.1 查看数据
# 查看前5行
print(df.head())
# 查看后5行
print(df.tail())
# 查看数据基本信息
print(df.info())
# 查看统计摘要
print(df.describe())
# 查看数据形状(行数,列数)
print(df.shape)
3.2 数据选择
# 选择单列
names = df['Name']
# 选择多列
subset = df[['Name', 'Age']]
# 通过标签选择行
row = df.loc[1] # 选择索引为1的行
# 通过位置选择行
row = df.iloc[1] # 选择第2行(从0开始)
# 通过标签选择多行
rows = df.loc[1:3] # 选择索引1到3的行(包含3)
# 通过位置选择多行
rows = df.iloc[1:3] # 选择第2到第3行(不包含第4行)
# 布尔索引
adults = df[df['Age'] > 25]
4. 数据清洗
数据清洗是数据分析中最重要的步骤之一,通常占整个分析过程的60-80%时间。
4.1 处理缺失值
# 检查缺失值
print(df.isnull().sum())
# 删除包含缺失值的行
df_cleaned = df.dropna()
# 删除包含缺失值的列
df_cleaned = df.dropna(axis=1)
# 填充缺失值
df_filled = df.fillna(0) # 用0填充
df_filled = df.fillna(method='ffill') # 用前一个值填充
df_filled = df.fillna(method='bfill') # 1用后一个值填充
# 填充不同列的不同值
df_filled = df.fillna({
'Age': df['Age'].mean(),
'City': 'Unknown'
})
4.2 处理重复值
# 检查重复行
print(df.duplicated().sum())
# 删除重复行
df_unique = df.drop_duplicates()
# 删除特定列的重复行
df_unique = df.drop_duplicates(subset=['Name'])
# 保留最后一个重复项
df_unique = df.drop_duplicates(subset=['Name'], keep='last')
4.3 数据类型转换
# 转换列的数据类型
df['Age'] = df['Age'].astype(int)
# 转换日期列
df['Date'] = pd.to_datetime(df['Date'])
# 转换为分类类型(节省内存)
df['Category'] = df['Category'].astype('category')
5. 数据转换与操作
5.1 添加/删除列
# 添加新列
df['Age_Group'] = df['Age'] // 10 * 10
# 基于条件创建列
df['Status'] = np.where(df['Age'] > 30, 'Senior', 'Junior')
# 删除列
df = df.drop('Age_Group', axis=1)
# 或者
del df['Age_Group']
5.2 apply函数
# 使用apply函数进行复杂转换
def calculate_age_group(age):
if age < 20:
return 'Teenager'
elif age < 35:
return 'Young Adult'
else:
return 'Adult'
df['Age_Group'] = df['Age'].apply(calculate_age_group)
# 使用lambda函数
df['Name_Length'] = df['Name'].apply(lambda x: len(x))
5.3 向量化操作
# 向量化操作比apply更快
# 好的做法
df['Age_In_10_Years'] = df['Age'] + 10
# 避免的做法(使用apply)
# df['Age_In_10_Years'] = df['Age'].apply(lambda x: x + 10)
6. 分组与聚合
6.1 groupby操作
# 基本分组
grouped = df.groupby('City')
# 多列分组
grouped = df.groupby(['City', 'Age_Group'])
# 聚合函数
agg_result = df.groupby('City')['Age'].agg(['mean', 'sum', 'count'])
# 多列不同聚合函数
agg_result = df.groupby('City').agg({
'Age': ['mean', 'max'],
'Name': 'count'
})
6.2 转换与过滤
# 转换(transform)
df['City_Mean_Age'] = df.groupby('City')['Age'].transform('mean')
# 过滤(filter)
filtered = df.groupby('City').filter(lambda x: x['Age'].mean() > 30)
7. 合并与连接
7.1 concat
# 垂直合并
df1 = pd.DataFrame({'A': [1, 2], 'B': [3, 4]})
df2 = pd.DataFrame({'A': [5, 6], 'B': [7, 8]})
result = pd.concat([df1, df2], axis=0) # 垂直合并
# 水平合并
result = pd.concat([df1, df2], axis=1) # 水平合并
7.2 merge
# 创建示例数据
left = pd.DataFrame({
'key': ['K0', 'K1', 'K2', 'K3'],
'A': ['A0', 'A1', 'A2', 'A3'],
'B': ['B0', 'B1', 'B2', 'B3']
})
right = pd.DataFrame({
'key': ['K0', 'K1', 'K2', 'K3'],
'C': ['C0', 'C1', 'C2', 'C3'],
'D': ['D0', 'D1', 'D2', 'D3']
})
# 内连接
result = pd.merge(left, right, on='key')
# 左连接
result = pd.merge(left, right, on='key', how='left')
# 右连接
result = pd.merge(left, right, on='key', how='right')
# 外连接
result = pd.merge(left, right, on='key', how='outer')
8. 时间序列处理
8.1 时间索引
# 创建时间索引
dates = pd.date_range('20230101', periods=6)
df_time = pd.DataFrame(np.random.randn(6, 4), index=dates, columns=list('ABCD'))
# 重采样
monthly = df_time.resample('M').mean() # 月度重采样
weekly = df_time.resample('W').sum() # 周度重采样
8.2 时间偏移
# 时间偏移
from pandas.tseries.offsets import DateOffset
df_time.index = df_time.index + DateOffset(months=1) # 所有日期加1个月
9. 性能优化技巧
9.1 使用适当的数据类型
# 优化内存使用
def optimize_memory(df):
start_mem = df.memory_usage().sum() / 1024**2
for col in df.columns:
col_type = df[col].dtype
if col_type != object:
c_min = df[col].min()
c_max = df[col].max()
if str(col_type)[:3] == 'int':
if c_min > np.iinfo(np.int8).min and c_max < np.iinfo(np.int8).max:
df[col] = df[col].astype(np.int8)
elif c_min > np.iinfo(np.int16).min and c_max < npinfo(np.int16).max:
df[col] = df[col].astype(np.int16)
elif c_min > np.iinfo(np.int32).min and c_max < np.iinfo(np.int32).max:
df[col] = df[col].astype(np.int32)
elif c_min > np.iinfo(np.int64).min and c_max < np.iinfo(np.int64).max:
df[col] = df[col].astype(np.int64)
else:
if c_min > np.finfo(np.float32).min and c_max < np.finfo(np.float32).max:
df[col] = df[col].astype(np.float32)
else:
df[col] = df[col].astype(np.float64)
else:
df[col] = df[col].astype('category')
end_mem = df.memory_usage().sum() / 1024**2
print(f'Memory usage reduced from {start_mem:.2f} MB to {end_mem:.2f} MB')
return df
9.2 避免链式索引
# 避免链式索引(可能返回副本而不是视图)
# 不推荐
# df[df['Age'] > 25]['Status'] = 'Senior'
# 推荐
df.loc[df['Age'] > 25, 'Status'] = 'Senior'
9.3 使用query方法
# 使用query进行快速筛选(对于大数据集更高效)
result = df.query('Age > 25 and City == "London"')
10. 高级技巧
10.1 窗口函数
# 滚动平均
df['Rolling_Mean'] = df['Age'].rolling(window=2).mean()
# 扩展窗口
df['Cumulative_Mean'] = df['Age'].expanding().mean()
# 指数加权移动平均
df['EWMA'] = df['Age'].ewm(span=2).mean()
10.2 透视表
# 创建透视表
pivot = pd.pivot_table(df,
values='Age',
index='City',
columns='Age_Group',
aggfunc='mean',
fill_value=0)
10.3 交叉表
# 创建交叉表
crosstab = pd.crosstab(df['City'], df['Age_Group'])
11. 实战案例:完整数据分析流程
让我们通过一个完整的案例来展示如何将这些技巧应用到实际工作中。
import pandas as pd
import numpy as np
# 1. 创建示例数据集
np.random.seed(42)
data = {
'CustomerID': range(1, 1001),
'Age': np.random.randint(18, 70, 1000),
'City': np.random.choice(['Beijing', 'Shanghai', 'Guangzhou', 'Shenzhen'], 1000),
'Purchase_Amount': np.random.uniform(10, 1000, 1000).round(2),
'Purchase_Date': pd.date_range('2023-01-01', periods=1000, freq='H'),
'Category': np.random.choice(['Electronics', 'Clothing', 'Food', 'Books'], 1000)
}
df = pd.DataFrame(data)
# 2. 数据清洗
# 添加一些缺失值
df.loc[0:50, 'Age'] = np.nan
df.loc[100:150, 'City'] = np.nan
# 处理缺失值
df['Age'].fillna(df['Age'].mean(), inplace=True)
df['City'].fillna('Unknown', inplace=True)
# 3. 特征工程
# 创建年龄分组
df['Age_Group'] = pd.cut(df['Age'],
bins=[0, 25, 35, 45, 100],
labels=['Young', 'Adult', 'Middle', 'Senior'])
# 创建购买金额分组
df['Purchase_Level'] = pd.cut(df['Purchase_Amount'],
bins=[0, 100, 500, 1000],
labels=['Low', 'Medium', 'High'])
# 4. 数据分析
# 按城市和年龄组统计
analysis = df.groupby(['City', 'Age_Group']).agg({
'CustomerID': 'count',
'Purchase_Amount': ['mean', 'sum', 'std']
}).round(2)
# 5. 高级分析
# 计算每个客户的总购买金额
customer_summary = df.groupby('CustomerID').agg({
'Purchase_Amount': 'sum',
'Purchase_Date': ['min', 'max'],
'City': 'first'
}).round(2)
# 计算购买频率
customer_summary['Purchase_Frequency'] = (
customer_summary[('Purchase_Date', 'max')] -
customer_summary[('Purchase_Date', 'min')]
).dt.days + 1
# 6. 结果导出
analysis.to_csv('city_age_analysis.csv')
customer_summary.to_excel('customer_summary.xlsx')
print("分析完成!")
print(f"数据集大小: {df.shape}")
print(f"处理后的数据:\n{df.head()}")
12. 常见问题与解决方案
12.1 SettingWithCopyWarning
# 问题:链式索引可能导致警告
# 不推荐
# df[df['Age'] > 25]['Status'] = 'Senior'
# 解决方案1:使用loc
df.loc[df['Age'] > 25, 'Status'] = 'Senior'
# 解决方案2:使用copy()
df_filtered = df[df['Age'] > 25].copy()
df_filtered['Status'] = 'Senior'
12.2 内存不足
# 分块读取大文件
chunk_size = 10000
chunks = pd.read_csv('large_file.csv', chunksize=chunk_size)
for chunk in chunks:
# 处理每个块
processed_chunk = chunk[chunk['value'] > 100]
# 保存或进一步处理
12.3 性能瓶颈
# 使用numba加速(对于数值计算)
from numba import jit
@jit(nopython=True)
def calculate_distance(x, y):
return np.sqrt(x**2 + y**2)
# 使用向量化代替循环
# 不推荐
# for i in range(len(df)):
# df.loc[i, 'distance'] = calculate_distance(df.loc[i, 'x'], df.loc[i, 'y'])
# 推荐
df['distance'] = calculate_distance(df['x'].values, df['y'].values)
13. 最佳实践总结
- 始终使用向量化操作:避免使用循环和apply,除非必要
- 优化数据类型:使用适当的数据类型可以大幅减少内存使用
- 使用loc/iloc:明确指定选择方式,避免链式索引
- 链式操作:合理使用链式操作可以使代码更简洁
- 分块处理大数据:对于超大数据集,使用chunksize参数
- 及时释放内存:处理完大数据后,使用del和gc.collect()释放内存
- 使用query方法:对于复杂筛选条件,query方法更高效
- 利用groupby的灵活性:agg、transform、filter各有用途
- 保持索引一致性:在合并操作前确保索引正确
- 文档化代码:为复杂的数据处理步骤添加注释
14. 总结
Pandas是一个功能强大的数据分析工具,掌握其高效使用方法可以显著提升数据分析的效率。本文从基础到高级,详细介绍了Pandas的核心功能和最佳实践。记住,高效使用Pandas的关键在于:
- 理解数据结构
- 掌握向量化操作
- 合理使用内存
- 熟练运用分组和聚合
- 避免常见陷阱
通过不断练习和实践,你将能够熟练运用Pandas解决各种数据分析问题。
