编写Python函数实现DataFrame指定列与整数相乘
处理DataFrame指定列数值的函数实现
原始DataFrame示例
| Date | B | C |
|---|---|---|
| 2022-07-24 | 12 | 1234 |
| 2021-02-01 | 13 | 6789 |
| 2020-04-30 | 14 | 4324 |
实现代码
import pandas as pd def scale_columns(df): # 创建副本避免修改原始数据 modified_df = df.copy() # 对B列所有数值乘以2 modified_df['B'] *= 2 # 对C列所有数值乘以3 modified_df['C'] *= 3 return modified_df
代码说明
- 先导入pandas库(如果你的环境还没导入的话)
- 函数接收目标DataFrame作为参数,先复制原DataFrame,这样操作不会影响原始数据;如果不需要保留原数据,可以直接操作传入的
df,省略复制步骤 - 使用
*=运算符直接对整列进行元素级乘法,pandas会自动遍历列内所有数值完成计算,高效处理1000行数据完全没问题 - 返回修改后的DataFrame
处理后示例结果
| Date | B | C |
|---|---|---|
| 2022-07-24 | 24 | 3702 |
| 2021-02-01 | 26 | 20367 |
| 2020-04-30 | 28 | 12972 |
内容的提问来源于stack exchange,提问作者Madness
相关产品推荐
相关产品推荐

