我的问题的前提是我想使用一个数据框(peergroups)创建与其他股票相同的股票组,然后计算另一个数据框(fun_data)的平均值,但我不知道如何使用一个数据框按年份和股票代码创建组,然后应用组,找到多个列的平均值,并在另一个数据帧中为这些平均值创建新列。感谢您的帮助。我目前掌握的数据如下
我从两个数据框开始,一个是基本数据,另一个是每年的公司同行组
趣味数据
import numpy as np
import pandas as pd
fun_data = {{'data': [12/30/1983, 12/30/1983, 12/30/1983, 12/30/1983, 12/30/1983, 12/30/1983, 12/30/1983, 12/30/1983, 12/30/1983, 12/30/1983, 12/30/1983, 12/30/1983, 12/30/1983, 12/30/1983, 12/30/1983, 1/3/1984, 1/3/1984, 1/3/1984, 1/3/1984, 1/3/1984, 1/3/1984, 1/3/1984, 1/3/1984, 1/3/1984, 1/3/1984, 1/3/1984, 1/3/1984, 1/3/1984, 1/3/1984, 1/3/1984],
'ticker': ['AA', 'KO', 'AMB', 'AMX', 'AR', 'AS', 'BUD', 'CLF', 'CRS', 'DOC', 'EC', 'EFU', 'FTX', 'HM', 'RJR', 'AA', 'KO', 'AMB', 'AMX', 'AR', 'AS', 'BUD', 'CLF', 'CRS', 'DOC', 'EC', 'EFU', 'FTX', 'HM', 'RJR'],
'mkt_cap': [10382076219, 28615981356, 89124668974, 96863568587, 69017311359, 71368368637, 36604633897, 91086629072, 87580223715, 70605054110, 93225158261, 91412455851, 76327466814, 60245266890, 33751408249, 92924687267, 97193082284, 43372080824, 94712408349, 60356743279, 32484886660, 18571138143, 64690517329, 24838868675, 23278782495, 34286838121, 46008417484, 24020283962, 3560654158, 79189294007],
'pe_ratio': [15, 24, 15, 20, 22, 19, 16, 22, 18, 13, 18, 16, 14, 24, 15, 12, 18, 22, 16, 21, 20, 16, 24, 18, 15, 24, 24, 18, 13, 18],}
df1 = pd.DataFrame(data=fun_data)
df1
对等群
import numpy as np
import pandas as pd
peergroup = {'year': [1983, 1983, 1983, 1983, 1983, 1984, 1984, 1984, 1984, 1984, 1983, 1983, 1983, 1984, 1984, 1984],
'ticker': ['AA', 'AA', 'AA', 'AA', 'AA', 'AA', 'AA', 'AA', 'AA', 'AA', 'KO', 'KO', 'KO', 'KO', 'KO', 'KO'],
'peer': [AMX, AS, CLF, CRS, EFU, HM, AMX, AR, EC, FTX, AMB, BUD, DOC, AMB, BUD, RJR]}
df2 = pd.DataFrame(data=peergroup)
df2
一旦我有了这些数据帧,我想象代码会执行这些步骤(如果有更好的方法,请随意调整)
我想你需要
merge
。首先要创建每个组的平均值,您可以在从日期和股票代码创建的列year上以及在df2 year和peer中创建merge
df1。然后在此合并数据帧中按df2中的tocker列分组:请注意,我没有得到与您预期输出相同的值,但我不确定您是如何在1983年得到19.3的KO的,因为您知道在df2中,它与三个值均小于16的股票代码相关联,所以我认为这是您想要的
现在您只需要
merge
再次将其返回到df1,年份和股票代码是刚刚创建的df_u的索引,加上一些fillna
和drop
列年份相关问题 更多 >
编程相关推荐