欢迎访问宙启技术站
智能推送

使用Python编写一个简单的股票交易程序

发布时间:2023-12-04 11:20:59

以下是一个简单的股票交易程序的示例代码:

class Stock:
    def __init__(self, name, price, quantity):
        self.name = name
        self.price = price
        self.quantity = quantity

class Portfolio:
    def __init__(self):
        self.stocks = {}

    def buy_stock(self, stock_name, price, quantity):
        if stock_name in self.stocks:
            self.stocks[stock_name].quantity += quantity
        else:
            stock = Stock(stock_name, price, quantity)
            self.stocks[stock_name] = stock

    def sell_stock(self, stock_name, price, quantity):
        if stock_name in self.stocks:
            if self.stocks[stock_name].quantity > quantity:
                self.stocks[stock_name].quantity -= quantity
            else:
                del self.stocks[stock_name]

    def portfolio_value(self):
        total_value = 0
        for stock_name, stock in self.stocks.items():
            total_value += stock.price * stock.quantity
        return total_value


# 使用例子
portfolio = Portfolio()

# 购买股票
portfolio.buy_stock("AAPL", 150, 10)
portfolio.buy_stock("GOOG", 1200, 5)

# 卖出部分股票
portfolio.sell_stock("AAPL", 160, 5)

# 查看持仓市值
print("持仓市值:", portfolio.portfolio_value())

在以上代码中,我们定义了两个类:股票(Stock)和投资组合(Portfolio)。

- 股票类具有名称、价格和数量属性。

- 投资组合类包含一个字典来存储持有的股票,其中键是股票名称,值是股票对象。

- 投资组合类提供了买入和卖出股票的方法,并根据股票的价格和数量更新股票持有数量。

- 此外,投资组合类还提供了计算持仓市值的方法,通过遍历所有的股票并乘以相应的股票价格计算总金额。

在使用例子中,我们创建了一个投资组合对象(portfolio),然后通过买入股票的方法将股票添加到投资组合中,然后卖出部分股票,并最后计算持仓市值并打印出来。

这只是一个简单的股票交易程序示例,实际的交易系统可能需要更复杂的逻辑和功能。