DEV Community

Bahman Shadmehr
Bahman Shadmehr

Posted on

Unveiling Financial Insights: Visualizing Stock Data with Matplotlib and Seaborn

In the world of finance, understanding historical stock data is crucial for making informed decisions. In this blog post, we'll leverage the power of Python libraries, specifically yfinance, Matplotlib, Seaborn, and Plotly, to fetch, analyze, and visualize stock data. We'll focus on a stock symbol, such as 'AAPL' (Apple Inc.), and create an array of visualizations to uncover key financial insights.

Fetching and Exploring Data

Let's begin by fetching historical stock data using the yfinance library and exploring the dataset.

import yfinance as yf

# Fetch historical stock data
symbol = 'AAPL'
stock_data = yf.download(symbol, start='2022-01-01', end='2023-01-01', progress=False)

# Display the first few rows of the dataset
print(stock_data.head())
Enter fullscreen mode Exit fullscreen mode

This sets the stage for our visual exploration of financial insights.

Visualizations for Financial Insights

1. Line Plot of Closing Prices

import matplotlib.pyplot as plt

# Line plot of closing prices
plt.figure(figsize=(12, 6))
plt.plot(stock_data['Close'], label=f'{symbol} Closing Price', linewidth=2)
plt.title(f'{symbol} Closing Prices Over Time')
plt.xlabel('Date')
plt.ylabel('Closing Price (USD)')
plt.legend()
plt.show()
Enter fullscreen mode Exit fullscreen mode

A simple line plot showcasing the closing prices over time, providing a fundamental overview of the stock's performance.

2. Distribution of Daily Returns

import seaborn as sns

# Seaborn style set
sns.set(style="whitegrid")

# Distribution of Daily Returns
plt.figure(figsize=(12, 6))
sns.histplot(stock_data['Adj Close'].pct_change().dropna(), bins=30, kde=True, color='blue')
plt.title(f'Distribution of {symbol} Daily Returns')
plt.xlabel('Daily Return')
plt.ylabel('Frequency')
plt.show()
Enter fullscreen mode Exit fullscreen mode

A histogram using Seaborn to display the distribution of daily returns, offering insights into the volatility of the stock.

3. Candlestick Chart

import plotly.graph_objects as go

# Candlestick chart
candlestick = go.Figure(data=[go.Candlestick(x=stock_data.index,
                                              open=stock_data['Open'],
                                              high=stock_data['High'],
                                              low=stock_data['Low'],
                                              close=stock_data['Close'])])

candlestick.update_layout(title=f'{symbol} Candlestick Chart',
                          xaxis_title='Date',
                          yaxis_title='Stock Price (USD)',
                          xaxis_rangeslider_visible=False)

candlestick.show()
Enter fullscreen mode Exit fullscreen mode

A detailed candlestick chart created with Plotly, providing a comprehensive view of open, high, low, and close prices.

4. Moving Average Plot

# Moving Average Plot
plt.figure(figsize=(12, 6))
stock_data['Close'].plot(label=f'{symbol} Closing Price', linewidth=2)
stock_data['Close'].rolling(window=30).mean().plot(label=f'{symbol} 30-Day Avg', linestyle='--', color='orange')
plt.title(f'{symbol} Closing Prices with 30-Day Moving Average')
plt.xlabel('Date')
plt.ylabel('Closing Price (USD)')
plt.legend()
plt.show()
Enter fullscreen mode Exit fullscreen mode

A 30-day moving average plot added to the closing prices to identify trends and smoothen short-term fluctuations.

5. Volume Plot

# Volume Plot
plt.figure(figsize=(12, 6))
plt.bar(stock_data.index, stock_data['Volume'], color='green', alpha=0.7)
plt.title(f'{symbol} Trading Volume Over Time')
plt.xlabel('Date')
plt.ylabel('Volume')
plt.show()
Enter fullscreen mode Exit fullscreen mode

A bar plot depicting the trading volume over time, offering insights into market activity levels.

6. Correlation Heatmap

# Correlation Heatmap
correlation_matrix = stock_data[['Open', 'High', 'Low', 'Close', 'Adj Close', 'Volume']].corr()
plt.figure(figsize=(10, 8))
sns.heatmap(correlation_matrix, annot=True, cmap='coolwarm', fmt=".2f", linewidths=0.5)
plt.title(f'Correlation Heatmap for {symbol} Financial Metrics')
plt.show()
Enter fullscreen mode Exit fullscreen mode

A heatmap using Seaborn to visualize the correlation matrix of various financial metrics, aiding in identifying relationships between different aspects of the stock data.

Conclusion

In this blog post, we've taken a journey through fetching, exploring, and visually analyzing financial data using Python. The combination of yfinance, Matplotlib, Seaborn, and Plotly provides a powerful toolkit for gaining key insights into stock performance. As you embark on your financial data analysis, feel free to adapt and customize these visualizations based on your specific needs and preferences. Happy analyzing!

Top comments (0)