Skip to content

Instantly share code, notes, and snippets.

@willy-r

willy-r/stock_buy_sell.py

Last active September 4, 2020 18:18
Show Gist options
  • Save willy-r/8a052d6684c6819cff5c51c91e054305 to your computer and use it in GitHub Desktop.
Save willy-r/8a052d6684c6819cff5c51c91e054305 to your computer and use it in GitHub Desktop.
Download ZIP
Given an array of numbers that represent stock prices (where each number is the price for a certain day), find 2 days when you should buy and sell your stock for the highest profit.
Raw
stock_buy_sell.py
def stock_buy_sell(prices: list):
"""Returns the 2 days when you should buy and sell the stock for the
highest profit.
"""
buy_day = prices.index(min(prices)) + 1
sell_day = prices.index(max(prices[buy_day:])) + 1
return f'Buy on day {buy_day}, sell on day {sell_day}'
if __name__ == '__main__':
prices = [110, 180, 260, 40, 310, 535, 695]
print(stock_buy_sell(prices))
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment