如何使用标准输入函数从数据框中切片日期？

Question

我在Indexing and selecting data 中看到了文档，其中涉及硬核脚本方法从数据帧中分割一系列数据。

import pandas as pd
import matplotlib.pyplot as plt

df = pd.read_csv('d1.csv')

df['time']=pd.to_datetime(df['time'], unit='ns')

df = df.drop('name', 1)

df['Time'] = df['time'].dt.time

df['date'] = df['time'].dt.date
df['date'] = pd.to_datetime(df['date'])

df = df.set_index(['date'])

df= df.loc['2018-07-04':'2018-07-05']

但是我需要从标准输入函数中选择一个数据范围，怎么做：

而不是在控制台的表格中使用df= df.loc['2018-07-04':'2018-07-05']say，它将被要求Enter the start date :和Enter the stop date :，这样我将只获得所选日期范围的数据。

我实际上是这样尝试的：

import pandas as pd
import matplotlib.pyplot as plt

df = pd.read_csv('d1.csv')

df['time']=pd.to_datetime(df['time'], unit='ns')

df = df.drop('name', 1)

df['Time'] = df['time'].dt.time

df['date'] = df['time'].dt.date
df['date'] = pd.to_datetime(df['date'])

df = df.set_index(['date'])
Starting_Date = input(" Please Enter the Starting_Date : ")
Ending_Date = input(" Please Enter the Ending_Date : ")

data = df[Starting_Date:Ending_Date]

但这不起作用……请看一下。

Answer 1

请试试这个。日期格式为year-month-day，例如“2018-08-16”。

from datetime import datetime

a = input('Starting_Date: ')
b = input('Ending_Date :')

starting_date = datetime.strptime(a, "%Y-%m-%d").date()
ending_date = datetime.strptime(b, "%Y-%m-%d").date()

df.loc[starting_date:ending_date]

希望对你有用:)