
正文
[Python] Read and plot data from csv file
提示:扫一扫查出行【扫一扫了解最新限行尾号】
复制提示
Install:
pip install pandas
pip install matplotlib # check out the doc from site
import pandas as pd
import matplotlib.pyplot as plt
from numpy import meandef load_df(symbol):
return pd.read_csv("data/{0}.csv".format(symbol))def get_max_close(symbol):
""" Return the maximum closing value for stock idicicated by symbol.
Note: Data for a stock is stored in file: data/<symbol>.csv
"""
df = load_df(symbol)
return df["Close"].max()def get_mean_volume(symbol):
df = load_df(symbol)
return mean(df['Volume'])def printAdjClose():
df = pd.read_csv('data/ibm.csv')
df[['Low', 'High']].plot()
plt.show()def test_run():
"""
Function called by Test Run
"""
for symbol in ['aapl', 'ibm']:
print("Max close")
print(symbol, get_max_close(symbol))
print("Mean Volume")
print(symbol, get_mean_volume(symbol))if __name__ == "__main__":
test_run()
printAdjClose()






