代码下载

点击此处下载代码


问题

如何将数据中，同一股票代码同一年的某个字段加总成一条？

我想把某公司同一年的数据var加总到一起


思路

可以通过pandas库实现这个需求

  1. 获取公司股票代码列表
  2. 获取某公司年份列表
  3. 对某个公司同年的var进行加总 (var代指一个字段或变量)
  4. for循环对所有的公司重复2-3操作

准备数据

import numpy as np
import pandas as pd

#强制股票代码转为str类型
df = pd.read_excel('data.xlsx', converters={"code": str})
df.head()