從 yfinance 操縱日期

Question

當我從 yfinance 提取股票數據時，我可以創建其他數據列來操縱“日期”列嗎？ 我是 python 的新手，仍然學到很多東西。 我已經使用股票價格數據創建了其他列，但我不知道如何操作“日期”列。

例如，2020 年 10 月 26 日，我想使用以下數據創建列：
day_of_week，星期一 = 1
年 = 2020
月 = 10
天 = 26
周 = 44
交易日 = 207

import pandas as pd
import numpy as np
import yfinance as yf
import pandas_datareader as pdr
import datetime as dt
import matplotlib.pyplot as plt

##Get stock price data
ticker = 'NVDA'

#Data time period
now = dt.datetime.now()
startyear = 2017
startmonth=1
startday=1
start = dt.datetime(startyear, startmonth, startday)

#get data from YFinance
df = pdr.get_data_yahoo(ticker, start, now)

#create a column
df['% Change'] = (df['Adj Close'] / df['Adj Close'].shift(1))-1

df['Range'] = df['High'] - df['Low']

df

Answer 1

你想使用你的 dataframe 的索引，它的類型是 pd.DatetimeIndex。 要將日期拆分為新列：

new_df = df.copy()
new_df['year'], new_df['month'], new_df['day'] = df.index.year, df.index.month, df.index.day

從第一個交易日開始進行算術運算：

start_date = df.index.min()
new_df['trade_day'] = df.index.day - start_date.day
new_df['trade_week'] = df.index.week - start_date.week
new_df['trade_year'] = df.index.year - start_date.year
new_df['day_of_week'] =  df.index.weekday
new_df['days_in_month'] =  df.index.days_in_month
new_df['day_name'] =  df.index.day_name()
new_df['month_name'] =  df.index.month_name()

選擇另一個開始日期

start_date = pd.to_datetime('2017-01-01')

Answer 2

我確實找出了大部分問題。 我無法弄清楚如何計算“交易日期”。

#Convert the 'Date' Index to 'Date' Column
df.reset_index(inplace=True)

#Create columns manipulating 'Date'
df['Year'] = df['Date'].dt.year
df['Month'] = df['Date'].dt.month
df['Day'] = df['Date'].dt.day
df['Week of Year'] = df['Date'].dt.isocalendar().week
df['Day of Week'] = df['Date'].dt.dayofweek