在函数间传递日期数据时，确保数据类型正确至关重要。 很多时候，从一个函数返回的日期数据可能被错误地格式化为字符串，导致在后续函数中使用时出现类型错误。本文将详细讲解如何解决这个问题，并提供示例代码。
问题分析
问题代码中，oi_data函数负责从API获取期权数据，并提取到期日期。原始代码将日期格式化为字符串，导致后续的fut_data函数无法正确处理。fut_data函数期望接收一个datetime.date对象作为expiry_date参数，但实际上接收到的是字符串。
解决方案
正确的做法是在oi_data函数中，直接返回datetime.date对象，而不是格式化后的字符串。这样可以避免类型转换带来的问题。
以下是修改后的代码：
from jugaad_data.nse import derivatives_df
from datetime import timedelta, date
from datetime import datetime
import pandas as pd
import requests
import json
def oi_data(Symbols):
baseurl = "https://www.nseindia.com/"
url = f'https://www.nseindia.com/api/option-chain-equities'
headers = {'user-agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, ''like Gecko) ''Chrome/80.0.3987.149 Safari/537.36','accept-language': 'en,gu;q=0.9,hi;q=0.8', 'accept-encoding': 'gzip, deflate, br'}
session = requests.Session()
request = session.get(baseurl, headers=headers, timeout=30)
cookies = dict(request.cookies)
params=[('symbol', Symbols)]
res = session.get(url, headers=headers, params=params, cookies=cookies, timeout=30)
res_text = res.text
data = json.loads(res_text)
exp_dt1 = data["records"]["expiryDates"][0]
datetime_obj = datetime.strptime(exp_dt1, '%d-%b-%Y')
exp_dt = datetime_obj.date() # 返回 datetime.date 对象
print(exp_dt)
return exp_dt
def fut_data(Symbols, exp_dt): # 接收 exp_dt 参数
ticker_df = derivatives_df(symbol=Symbols, from_date = date.today() - timedelta(days = 30), to_date= date.today(),expiry_date=exp_dt, instrument_type="FUTSTK")
fut_df = ticker_df
print(fut_df)
return fut_df
Symbols = ['MARUTI']
for symbol in Symbols:
exp_dt = oi_data(symbol)
fut_df = fut_data(symbol, exp_dt) # 传递 exp_dt 参数代码解释
-
oi_data 函数修改：
- 将exp_dt = datetime_obj.strftime("%Y, %m, %d")替换为exp_dt = datetime_obj.date()。这样，exp_dt变量将存储一个datetime.date对象，而不是一个格式化的字符串。
-
fut_data 函数修改：
- 修改函数定义，使其接受exp_dt作为参数：def fut_data(Symbols, exp_dt):
-
主程序修改：
- 在调用fut_data函数时，将exp_dt作为参数传递：fut_df = fut_data(symbol, exp_dt)
注意事项
- 确保在函数间传递日期数据时，始终保持数据类型的一致性。
- 在需要格式化日期时，最好在最终显示或存储时进行格式化，而不是在函数内部进行。
- 使用datetime.date对象进行日期计算和比较，可以避免字符串转换带来的性能损耗。
- derivatives_df函数需要传递一个datetime.date对象，而不是字符串。
总结
通过以上修改，可以有效地解决日期格式化问题，确保fut_data函数能够正确接收和处理日期数据。 核心在于理解不同函数对数据类型的要求，并在函数间传递数据时保持数据类型的一致性。 在编写涉及日期处理的Python代码时，务必注意数据类型，避免不必要的类型转换，从而提高代码的可读性和可维护性。
