使用【python】写编写一个函数,输入参数为股票代码和公司简称,返回值为对应公司的所属证券交易所、所属板块以及是否为 ST公司。其中,编写异常处理代码,当代码执行不成功时,打印提示信息“无法识别,请检查输入信息是否正确”。分别用正确和错误的函数输入测试编写的函数。
时间: 2024-09-21 08:04:45 浏览: 39
《量化交易入门(六)Python开发-函数和调用》配套源码
在Python中,你可以创建一个函数`get_company_info`来实现这个功能。首先,你需要模拟一个数据库或API查询来获取数据,这里我们假设数据是一个字典或者存储在一个列表中,包含了股票代码、交易所、板块和ST状态等信息。由于实际网络请求通常会引发异常,我们将使用try-except块来捕获并处理这些异常。
```python
def get_company_info(stock_code, company_abbr):
# 假设有一个字典数据源,实际应用中可能是从数据库或API获取
data_source = {
'AAPL': {'exchange': 'NASDAQ', 'sector': 'Technology', 'st_status': False},
'TSLA': {'exchange': 'NASDAQ', 'sector': 'Automotive', 'st_status': True},
# 更多股票数据...
}
try:
# 检查输入的代码是否在数据源中
if stock_code not in data_source:
raise ValueError("无法识别,请检查输入信息是否正确")
# 获取并返回公司信息
exchange = data_source[stock_code]['exchange']
sector = data_source[stock_code]['sector']
st_status = data_source[stock_code]['st_status']
return exchange, sector, st_status
except ValueError as e:
print(e)
# 测试函数
print(get_company_info('AAPL', 'Apple')) # 正确输入,应返回 NASDAQ, Technology, False
print(get_company_info('XYZ', 'Company XYZ')) # 错误输入,应打印 "无法识别,请检查输入信息是否正确"
```
阅读全文