如何从多个函数中添加迭代

Question

outstanding_shares = get_class_i(f) + get_class_c(f) + get_class_i(f)

我有点笨。我试图理解如何一次又一次地将这些函数加在一起。相反，我只得到一个大的输出。其想法是在一个月内逐日获得流通股，最终结果是一个线图。

def get_class_a(d):
    cs = 0

    excel_file = 'nav/Catalyst_Exceed_Port_daily_nav' + d +'.xls'
    wb = xlrd.open_workbook(excel_file, logfile=open(os.devnull, 'w'))
    df = pd.read_excel(wb)

    for index, row in df.iterrows():

        if (row[0] == 'CATALYST/EXCEED DEFINED SHIELD FUND Class A'):
            cs = row[7]
    return cs

c = 0
for f in ('20200102', '20200103', '20200106', '20200107', '20200108', '20200109', '20200110',
          '20200113', '20200114', '20200115', '20200116', '20200117', '20200121', '20200122',
          '20200123', '20200124', '20200127', '20200128', '20200129', '20200130', '20200131'):
    #print(get_class_a(f), end=", ")

    print()
#########Grabbing Class C Shares Outstanding
def get_class_c(d):
    cs = 0

    excel_file = 'nav/Catalyst_Exceed_Port_daily_nav' + d +'.xls'
    wb = xlrd.open_workbook(excel_file, logfile=open(os.devnull, 'w'))
    df = pd.read_excel(wb)

    for index, row in df.iterrows():

        if (row[0] == 'CATALYST/EXCEED DEFINED SHIELD FUND Class C'):
            cs = row[7]
    return cs

c = 0
for f in ('20200102', '20200103', '20200106', '20200107', '20200108', '20200109', '20200110',
          '20200113', '20200114', '20200115', '20200116', '20200117', '20200121', '20200122',
          '20200123', '20200124', '20200127', '20200128', '20200129', '20200130', '20200131'):
    #print(get_class_c(f), end=", ")

    print()
#########Grabbing Class I Shares Outstanding
def get_class_i(d):
    cs = 0

    excel_file = 'nav/Catalyst_Exceed_Port_daily_nav' + d +'.xls'
    wb = xlrd.open_workbook(excel_file, logfile=open(os.devnull, 'w'))
    df = pd.read_excel(wb)

    for index, row in df.iterrows():

        if (row[0] == 'CATALYST/EXCEED DEFINED SHIELD FUND Class I'):
            cs = row[7]
    return cs

c = 0
for f in ('20200102', '20200103', '20200106', '20200107', '20200108', '20200109', '20200110',
          '20200113', '20200114', '20200115', '20200116', '20200117', '20200121', '20200122',
          '20200123', '20200124', '20200127', '20200128', '20200129', '20200130', '20200131'):
    #print(get_class_i(f), end=", ")
    print()



outstanding_shares = get_class_i(f) + get_class_c(f) + get_class_i(f)

print(outstanding_shares, end=", ")

Answer 1

我会这样做:

import pandas

datestrings = [
    '20200102', '20200103', '20200106', '20200107', '20200108', '20200109', '20200110',
    '20200113', '20200114', '20200115', '20200116', '20200117', '20200121', '20200122',
    '20200123', '20200124', '20200127', '20200128', '20200129', '20200130', '20200131'
]

# compile all of the data
data = pandas.concat([
    pandas.read_excel('nav/Catalyst_Exceed_Port_daily_nav' + d +'.xls')
    for d in datestrings
], ignore_index=True)

# create all list of the valid values of the first column
classes_you_want = [
    f'CATALYST/EXCEED DEFINED SHIELD FUND Class {letter}'
    for letter in ('A', 'C', 'I')
]

# select all of the rows where the first column has a valid value,
# then select the 8th column, take the sum
outstanding_shares = data.loc[data.iloc[:, 0].isin(classes_you_want)].iloc[:, 7].sum()