使用 Python 用户输入更新值

Question

我有一个数据集，每当更新“更新”列中的日期值时，另一个 列也将更新。逻辑是：

Date1 is 3 months from Date2
Date2 is 1 month from Date3
and Update is 1 month from Date3

唯一变化的数据是日期，这些数据基本上会根据用户输入发生变化。

数据

Date1       Date2       Date3       Update
1/1/2021    4/1/2021    5/1/2021    6/1/2021
5           2           1           1
        

希望

输入提示将询问用户他们希望输入的日期值。 用户输入日期“2021 年 8 月 1 日”，这会更新剩余列的日期值。

Date1       Date2       Date3       Update
3/1/2021    6/1/2021    7/1/2021    8/1/2021
5           2           1           1

正在做

我相信我可以结合使用函数和用户提示来解决这个问题。

#take input
datevalue = input("Enter date value: ")
print(datevalue)




#use input variable in function or script to create date update
s = df['Update'].str.replace(r'(\S+) (\S+)', r'\2\1')
df['Update'] = (pd.PeriodIndex(s, freq='D') + 3).strftime('D%q %Y')

我正在寻找关于如何最好地解决此问题的一些起点建议或良好的基础/文档。我还在研究。任何建议表示赞赏。

Answer 1

你的数据格式有点乱，但这应该适合你：

datevalue = pd.to_datetime(input("Enter date value: "))

df = df.T
df[1] = df[1].astype(int)
df.loc['Update', 1] = 0
df[0] = df[1].apply(lambda x: datevalue - pd.DateOffset(months=x))
df = df.T

输出：

>>> df
                 Date1                Date2                Date3               Update
0  2021-03-01 00:00:00  2021-06-01 00:00:00  2021-07-01 00:00:00  2021-08-01 00:00:00
1                    5                    2                    1                    0