利用Python实现简单的Excel统计函数【python爬虫】

目录

• 需求分析
• 解决步骤
• 最终结果
• 技术总结

import pandas as pd
#创建一个计算数据的函数
def calculate(df):
  pass
#读取原始数据，将索引列去除
df = pd.read_excel("需求0621.xlsx",index_col=0)
#调用计算数据的函数
calculate(df)

#计算累计和
    lst1 = []
    sum = 0
    for i in range(df.shape[0]):
        if i == 0:
            lst1.append(df["N"][i])
            sum += df["N"][i]
        else:
            sum += df["N"][i]
            lst1.append(sum)
    df["累计和"] = lst1

#计算回撤
    lst2 = []
    max = 0
    for i in range(df.shape[0]):
        if i == 0:
            lst2.append(0)
        elif df["累计和"][i] > max:
            max = df["累计和"][i]
            lst2.append(0)
        elif df["累计和"][i] 

#计算连续正确的个数
    lst3 = []
    correct = 0
    for i in range(df.shape[0]):
        if df["N"][i] >= 0:
            correct += 1
            lst3.append(correct)
        else:
            lst3.append(0)
            correct = 0
    df["连续正确"] = lst3

#计算连续错误的个数
    lst4 = []
    mistake = 0
    for i in range(df.shape[0]):
        if df["N"][i] <0:
            mistake += 1
            lst4.append(mistake)
        else:
            lst4.append(0)
            mistake = 0
    df["连续错误"] = lst4

#计算连续正确值
    lst5 = []
    for i in range(df.shape[0]):
        lst5.append("")
    right = 0
    for i in range(df.shape[0]):
        if df["连续正确"][i] != 0:
            right += df["N"][i]
        elif df["连续正确"][i] == 0 and right != 0:
            lst5[i-1] = right
            right = 0
    df["连续正确值"] = lst5

#计算连续错误值
    lst6 = []
    for i in range(df.shape[0]):
        lst6.append("")
    wrOng= 0
    for i in range(df.shape[0]):
        if df["连续错误"][i] != 0:
            wrong += df["N"][i]
        elif df["连续错误"][i] == 0 and wrong != 0:
            lst6[i-1] = wrong
            wrOng= 0
    df["连续错误值"] = lst6

    df.to_excel("完成计算.xlsx")
    print("保持成功")