上一篇文章,我们只是简单把数据放到python里。还有好多问题需要慢慢优化。例如我们可以把股票名字放在函数参数里,这样我们可以更加方便地读取我们想要的公司数据。
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的用法,就是可以把相同格式的语句,免去重复使用的麻烦。