上一篇文章,我們只是簡單把數(shù)據(jù)放到python里。還有好多問題需要慢慢優(yōu)化。例如我們可以把股票名字放在函數(shù)參數(shù)里,這樣我們可以更加方便地讀取我們想要的公司數(shù)據(jù)。
import pandas as pd
def get_max_close(symbol):
df = pd.read_csv("data/{}.csv".format(symbol))
return df['Close'].max()
def test_run():
for symbol in ['AApl','IBM']:
print 'Max Close'
print symbol,get_max_close(symbol)
if __name__=="__main__":
test_run()

輸出結(jié)果
這段代碼,有一個(gè)format的用法,就是可以把相同格式的語句,免去重復(fù)使用的麻煩。