使用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),然后通过买入股票的方法将股票添加到投资组合中,然后卖出部分股票,并最后计算持仓市值并打印出来。
这只是一个简单的股票交易程序示例,实际的交易系统可能需要更复杂的逻辑和功能。
