上一篇文章,我們只是簡單把數據放到python里镊折。還有好多問題需要慢慢優(yōu)化揍魂。例如我們可以把股票名字放在函數參數里税手,這樣我們可以更加方便地讀取我們想要的公司數據。
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()
這段代碼拯辙,有一個format的用法,就是可以把相同格式的語句,免去重復使用的麻煩吁脱。