Python计算权重累积降水量WAP

#!usr/bin/env python
# -*- coding:utf-8 -*-
"""
@author: Suyue
@file: calculateWAP.py
@time: 2023/05/17
@desc:
"""
# python实现对Excel某一列乘以某个常数,结果存放在另一个单元格中
import pandas as pd
import math


#读取数据
data = pd.read_excel('G:/drought/processdata/datas/rainmonth.xlsx',sheet_name='50525')

#提取rain用于WAP计算
rain = data["rain"].values
#进行WAP公式计算
WAP1 = 0.85 * rain
WAP2 = math.pow(0.85,2)*rain

#第一列第二行加第二列第一行
a = WAP1[1:]
b = WAP2[:383]
print(data.head())
#错位相加
WAP3 = a + b
print(WAP3)


# 将结果保存到文件
OUT = pd.DataFrame([WAP1, WAP2, WAP3],
                   index = ['WAP1','WAP2','WAP3']).T
OUT.to_excel(r'G:/drought/processdata/datas/WAP.xlsx', index = False)

for i in range(len(df)-1):
    df.loc[i,'新列名'] = df.loc[i+1,'第一列列名'] + df.loc[i,'第二列列名']