密谋:如何仅在图形上显示今天的数据?

时间:2020-08-10 07:46:07

标签: python pandas datetime plotly

我使用X轴上的时间轴制作了一个图表,但是我只想显示今天的数据。 (即,将日期限制为当前日期)

这是我用来生成这些图形的代码:

fig = go.Figure()
fig = make_subplots(rows=2,cols=1,shared_xaxes=True,vertical_spacing=0.02)
fig.add_trace(go.Scatter(x=data['time'],y=data['x_last_recorded'],name='xyz',mode='lines+markers'),row=2,col=1)
fig.add_trace(go.Scatter(x=predict_data['time'],y=predict_data['x1_last_recorded'],name='x1yz',mode='lines'),row=2,col=1)

fig.update_layout(height=800,width=1500,title='first_graph',yaxis_title='Values')

这给了我一张我想要的图形,但是它显示了数据框中存在的所有日期。如何仅获取当前日期的数据? 时间的结构:dd-mm-yyyy hh:mm

1 个答案:

答案 0 :(得分:1)

我们可以通过使用df_current = df[df.index.date==df.index.date[-1]]之类的方法为数据框设置子集来解决您的挑战,然后通过让下拉菜单中的不同选项代表不同的子集来重新设置图形的样式。

以下是不同子集/选择选项的结果图:

所有日期

enter image description here

当前日期

enter image description here

完整代码:

# imports
import plotly.graph_objects as go
import pandas as pd
import numpy as np

# sample data in the form of an hourlt
np.random.seed(1234)
tseries = pd.date_range("01.01.2020", "01.04.2020", freq="H")
data = np.random.randint(-10, 12, size=(len(tseries), 2))
df = pd.DataFrame(index=tseries, data=data)
df.drop(df.tail(1).index,inplace=True)
df.columns=list('AB')
df.iloc[0]=0
df=df.cumsum()

# subset method
df_current = df[df.index.date==df.index.date[-1]]

# plotly setup
fig = go.Figure()

# set up a trace for each column in a dataframe
for col in df.columns:
    fig.add_trace(go.Scatter(x=df.index, y =df[col], name=col))

# container for updatemenus and buttons
updatemenu = []
buttons = []

# button 1
buttons.append(dict(method='restyle',
                    label='All',
                    visible=True,
                    args=[{'y':[df[col].values for col in df.columns],
                           'x':[df.index],
                           'type':'scatter'}, ],))
# button 2
buttons.append(dict(method='restyle',
                    label='Current',
                    visible=True,
                    args=[{'y':[df_current[col].values for col in df_current.columns],
                           'x':[df_current.index],
                           'type':'scatter'}, ],))

# menu setup
your_menu = dict()
updatemenu.append(your_menu)

updatemenu.append(your_menu)
updatemenu[0]['buttons'] = buttons
updatemenu[0]['direction'] = 'down'
updatemenu[0]['showactive'] = True

# add dropdown menus to the figure
fig.update_layout(showlegend=False, updatemenus=updatemenu)
fig.show()