编写函数,使用open()函数读取股票文件"000001.csv"的日期"trade_date"和收盘价"close"两列交易数据,将日期作为"键",收盘价作为"值"存为一个字典对象。
时间: 2024-06-23 20:01:54 浏览: 119
在Python中,你可以使用内置的`csv`模块来读取CSV文件,然后创建一个字典,其中日期作为键,收盘价作为值。以下是一个简单的示例,展示了如何实现这个功能:
```python
import csv
def read_stock_data(filename):
stock_dict = {} # 创建一个空字典来存储数据
with open(filename, 'r') as file: # 使用open函数打开文件
reader = csv.DictReader(file) # 使用DictReader读取文件,每一行会变成一个字典,方便按键操作
for row in reader: # 遍历文件中的每一行
trade_date = row['trade_date'] # 获取日期字段
close_price = float(row['close']) # 将收盘价字段转换为浮点数(假设它是数值类型)
stock_dict[trade_date] = close_price # 将数据添加到字典中
return stock_dict
# 示例用法
stock_data = read_stock_data('000001.csv')
阅读全文