用python绘制多重价格图

2024-09-29 23:27:19 发布

您现在位置:Python中文网/ 问答频道 /正文

我在udacity(我是新手)学习,我坚持这一点。我复制代码并运行,但没有输出

这是udacity链接https://www.youtube.com/watch?v=vmF6iEQzC2A

*我的csv文件与视频不同

here is my csv link

这是我的密码

    """Slice and plot"""

import os
import pandas as pd
import matplotlib.pyplot as plt

def plot_selected(df, columns, start_index, end_index):
    plot_data(df.ix[start_index:end_index,columns],title="Selected Data")

def symbol_to_path(symbol, base_dir="data"):
    """Return CSV file path given ticker symbol."""
    return os.path.join(base_dir, "{}.csv".format(str(symbol)))


def get_data(symbols, dates):
    """Read stock data (adjusted close) for given symbols from CSV files."""
    df = pd.DataFrame(index=dates)
    if 'EURUSDCSV' not in symbols:  # add EUR for reference, if absent
        symbols.insert(0, 'EURUSDCSV')

    for symbol in symbols:
        df_temp = pd.read_csv(symbol_to_path(symbol), index_col='DateTime',
                              parse_dates=True, usecols=['DateTime', 'Close'], na_values=['nan'])
        df_temp = df_temp.rename(columns={'Close': symbol})
        df = df.join(df_temp)
        if symbol == 'EURUSDCSV':  # drop dates SPY did not trade
            df = df.dropna(subset=["EURUSDCSV"])

    return df


def plot_data(df, title="Stock prices"):
    """Plot stock prices with a custom title and meaningful axis labels."""
    ax = df.plot(title=title, fontsize=12)
    ax.set_xlabel("Date")
    ax.set_ylabel("Price")
    plt.show()


def test_run():
    # Define a date range
    dates = pd.date_range('2015-01-01', '2015-12-31')

    # Choose stock symbols to read
    symbols = ['EURUSDCSV', 'USDJPYcsv']  # SPY will be added in get_data()

    # Get stock data
    df = get_data(symbols, dates)

    # Slice and plot
    plot_selected(df, ['EURUSDCSV', 'USDJPYcsv'], '2015-01-01', '2015-12-31')


if __name__ == "__main__":
    test_run()

Tags: csvpathdfdataindexifplottitle

热门问题