使用python3.9,将excel表格中根据列号指定几列的数值乘以设置的权值后相加,存到新的excel表格中
时间: 2024-02-13 19:59:40 浏览: 124
python读取excel指定列数据并写入到新的excel方法
5星 · 资源好评率100%
可以使用Python中的pandas库来实现此功能。以下是一个简单的示例代码:
```python
import pandas as pd
# 读取原始Excel文件
df = pd.read_excel('input.xlsx')
# 列号指定的列
cols = ['A', 'B', 'C']
# 设置的权值
weights = [0.5, 0.3, 0.2]
# 计算新列的值
df['new_col'] = df[cols].apply(lambda row: sum(row * weights), axis=1)
# 写入新的Excel文件
df.to_excel('output.xlsx', index=False)
```
在代码中,我们首先使用pandas库中的`read_excel`函数读取原始Excel文件。然后,我们指定需要计算的列号(例如'A', 'B', 'C'),以及这些列的权值(例如0.5, 0.3, 0.2)。接着,我们使用pandas中的`apply`函数计算新的列的值,并将其添加到DataFrame中。最后,我们使用pandas中的`to_excel`函数将结果写入新的Excel文件中。
注意,代码中的列号是指Excel表格中的列字母,而不是列索引。如果你想要根据列索引指定列,可以使用`iloc`函数代替`loc`函数。
阅读全文