GitHub Copilot实战 Leetcode和Alpha Vantage API获取股票数据

Posted 2023-03-13 AI架构师易筋

GitHub Copilot 可以提升编码速度25%。

需要在 visual studio code 添加插件 GitHub Copilot

1. Base Python 创建数组和Person class

# Create a list of 100 numbers.
numbers = [i for i in range(100)]

def get_even_numbers(numbers):
    """Return a list of even numbers from the given list."""
    return [i for i in numbers if i % 2 == 0]

even_numbers = get_even_numbers(numbers)
odd_numbers = [i for i in numbers if i not in even_numbers]

print("Even numbers: ".format(even_numbers))
print("Odd numbers: ".format(odd_numbers))

class Person:
    def __init__(self, name, age):
        self.name = name
        self.age = age

# A list of 10 people.
PEOPLE = [
    Person("John", 32),
    Person("Jane", 28),
    Person("Bob", 30),
    Person("Alice", 24),
    Person("Jill", 26),
    Person("Jack", 30),
    Person("Jen", 28),
    Person("Bill", 32),
    Person("Barbara", 24),
    Person("Jeff", 30),
]

2. Leetcode 转化罗马数字为阿拉伯数字

"""
    A program that converts a Roman numberal to an integer.
"""

ROMAN_NUMERAL = 
    "I": 1,
    "V": 5,
    "X": 10,
    "L": 50,
    "C": 100,
    "D": 500,
    "M": 1000,


def roman_to_int(roman_numeral: str) -> int:
    """
    Convert a Roman numeral to an integer.
    """
    result = 0
    for i, numeral in enumerate(roman_numeral):
        value = ROMAN_NUMERAL[numeral]
        if i + 1 < len(roman_numeral) and ROMAN_NUMERAL[roman_numeral[i + 1]] > value:
            result -= value
        else:
            result += value
    
    return result

TEST_CASES = [
    ("I", 1),
    ("II", 2),
    ("III", 3),
    ("IV", 4),
    ("V", 5),
    ("VI", 6),
    ("VII", 7),
    ("VIII", 8),
    ("IX", 9),
    ("X", 10),
    ("XI", 11),
    ("XII", 12),
    ("XIII", 13),
    ("XIV", 14),
    ("XV", 15),
    ("XVI", 16),
    ("XVII", 17),
    ("XVIII", 18),
    ("XIX", 19),
    ("XX", 20),
    ("XXI", 21)
]

for test_input, expected in TEST_CASES:
    result = roman_to_int(test_input)
    assert result == expected, f"Expected expected, got result"
    print(result, result == expected)

3. 获取股票数据Alpha Vantage API

https://www.alphavantage.co/documentation/
注册 api key
https://www.alphavantage.co/support/#api-key

"""
    Use Alpha Vantage API to get stock data and plot it in a graph.
 """

import requests
import pandas as pd
import matplotlib.pyplot as plt
 
API_KEY = "RTART83NP1RD7G8X"
STOCK_SYMBOL = "MSFT"
def get_daily_stock_data(symbol: str):
    """
    Get daily stock data for the given symbol.
    """
    url = f"https://www.alphavantage.co/query?function=TIME_SERIES_DAILY&symbol=symbol&apikey=API_KEY"
    response = requests.get(url)
    response.raise_for_status()
    data = response.json()
    
    return data

def get_intraday_stock_data(symbol: str):
    """
    Get intraday stock data for the given symbol.
    """
    url = f"https://www.alphavantage.co/query?function=TIME_SERIES_INTRADAY&symbol=symbol&interval=5min&apikey=API_KEY"
    response = requests.get(url)
    response.raise_for_status()
    data = response.json()
    
    return data

def main():
    data = get_intraday_stock_data(STOCK_SYMBOL)
    # print(data)
    # Turn the data into a pandas DataFrame.
    df = pd.DataFrame(data["Time Series (5min)"]).T
    df.index = pd.to_datetime(df.index)
    df = df.astype(float)

    # Print the dataframes.
    print(df.head())

    # Plot the data in a graph.
    plt.plot(df["4. close"])
    plt.show()

    pass

if __name__ == "__main__":
    main()

参考

https://www.youtube.com/watch?v=tG8PPne7ef0&ab_channel=pixegami