Random Candlestick Chart Generation with Seaborn

  • Share this:

Code introduction


This code defines a function named `random_candlestick_plot` that uses the `candlestick_ohlc` function from the seaborn library to generate a random candlestick chart of stock prices. The function accepts stock data, a date column, and a price column as inputs.


Technology Stack : The code uses the seaborn library and pandas for data manipulation. It also utilizes matplotlib for plotting the candlestick chart.

Code Type : The type of code

Code Difficulty : Advanced


                
                    
import seaborn as sns
import numpy as np
import pandas as pd

def random_candlestick_plot(data, x, y):
    """
    This function generates a random candlestick plot using seaborn's `candlestick_ohlc` function.
    """
    # Create a DataFrame from the provided data
    df = pd.DataFrame(data, columns=[x, 'Open', 'High', 'Low', 'Close'])
    
    # Generate a random color for the candles
    colors = ['green' if close > open else 'red' for open, close in zip(df['Open'], df['Close'])]
    
    # Plot the candlestick chart
    sns.candlestick_ohlc(df, x=x, y='Close', color=colors)
    
    plt.show()

# Example usage:
# data = {
#     'Date': ['2021-01-01', '2021-01-02', '2021-01-03'],
#     'Open': [100, 102, 101],
#     'High': [105, 107, 106],
#     'Low': [95, 98, 96],
#     'Close': [103, 104, 105]
# }
# random_candlestick_plot(data, 'Date', 'Close')