[英]Python iterate through list with pointer
I am trying to create a session template of dates for a dataframe in pandas based on the start and end day of the week of my given dataframe. 我有开始和结束日期的缩写(Mo、Tu、We 等)和开始/结束时间(上午 8:30、下午 5:30 等)。
我想要创建的是一个模板,它给出了开始日的缩写、跨越的时间和结束日。 例如,我的 dataframe 目前如下所示:
Start Time End Time Start/End Namestart Nameend Days Session Template
Mo 8:30 AM Th 5:30 PM Mo 8:30 AM-Th 5:30 PM Mo Th 4 Day 4 Day Mo 8:30 AM-Th 5:30 PM
We 8:30 AM Fr 12:30 PM We 8:30 AM-Fr 12:30 PM We Fr 3 Day 3 Day We 8:30 AM-Fr 12:30 PM
当前的 session 模板为我提供了天数、开始时间、结束时间以及它开始/结束的星期几。 但是,我希望它给出该项目跨越的每一天。 对于上面的示例,它应该产生:
4 Day Mo 8:30 AM-5:30 PM, Tu 8:30 AM-5:30 PM, We 8:30 AM-5:30 PM, Th 8:30 AM-5:30 PM.
3 Day We 8:30 AM-5:30 PM, Th 8:30 AM-5:30 PM, Fr 8:30 AM-12:30 PM
您可以这样做:
import pandas as pd
import re
import itertools
pd.set_option('display.max_columns', 100)
pd.set_option('display.width', 1000)
df = pd.read_csv("data.csv")
print(df, "\n")
days = ['Mo', 'Tu', 'We', 'Th', 'Fr', 'Sa', 'Su']
for index, row in df.iterrows():
# get the start and end days
start_day = row['Namestart']
end_day = row['Nameend']
# get the start end end times
start_time = re.findall(r'\s(\d+\:\d{2}\s?(?:AM|PM|am|pm))',
row['Start Time'])[0]
end_time = re.findall(r'\s(\d+\:\d{2}\s?(?:AM|PM|am|pm))',
row['End Time'])[0]
# get the indices corresponding to the start and end days
start_index = days.index(start_day)
end_index = days.index(end_day)+1
# count the number of days
cnt = end_index - start_index
print(cnt, "days\t", end='')
# slice the days list from start_index to end_index
for day in itertools.islice(days, start_index, end_index):
if (day!=end_day):
print(day, start_time, "- 5:30 PM\t", end='')
else:
print(day, start_time, "-", end_time, end='')
print() # to start a new line before printing each row
Output:
Start Time End Time Start/End Namestart Nameend Days Session Template
0 Mo 8:30 AM Th 5:30 PM Mo 8:30 AM-Th 5:30 PM Mo Th 4 Day 4 Day Mo 8:30 AM-Th 5:30 PM
1 We 8:30 AM Fr 12:30 PM We 8:30 AM-Fr 12:30 PM We Fr 3 Day 3 Day We 8:30 AM-Fr 12:30 PM
4 days Mo 8:30 AM - 5:30 PM Tu 8:30 AM - 5:30 PM We 8:30 AM - 5:30 PM Th 8:30 AM - 5:30 PM
3 days We 8:30 AM - 5:30 PM Th 8:30 AM - 5:30 PM Fr 8:30 AM - 12:30 PM
注释应该解释代码。 我使用的正则表达式的解释可以在这个答案中找到 - https://stackoverflow.com/a/49217300/6590393 。
另外,请注意上面的代码是基于你只在列表中前进的假设。 因此,例如,Sa-Mo 不会产生预期的结果。 如果您需要,我会留给您处理边界情况。
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.