在库存图中绘制特定行的问题

Question

我正在使用大熊猫在数据框架中使用DateTime索引绘制库存数据图。我想在df_raw [col_name]！= 0的行中绘制黑色标记/a>。当没有数据帧行满足条件时，图显示fine 。

您能帮我理解我在做错什么吗？请注意，该股票的数据已有几十年了，但我仅扣除了100万个数据的最后7天。

非常感谢！

fig.add_trace(go.Scatter(
    x=df_raw[df_raw[col_name]!=0].index,
    y=[df_raw['Close'][j] for j in df_raw[df_raw[col_name]!=0].index],
    mode='markers',
    marker=dict(
        size=8,
        color='black',
        symbol='cross'
    ),
    name='Cdl Pattern'
))

Answer 1

我可能已经找到了答案，并希望在此处发布它，以防它帮助他人。这是完整的绘图代码。请注意，它首先绘制“关闭”，然后绘制标记。

绘制“关闭”时，系统使用行号而不是索引，但是在绘制标记时，它明确指示使用索引。由于索引是DateTime，因此两个图最终都以不匹配的X轴。

fig = go.Figure()
fig.add_trace(go.Scatter(
    ###No x-axis provided, the system uses dataframe row numbers###
    y=validation_df['Close'],
    mode='lines+markers',
    name='Original Plot'
))

fig.add_trace(go.Scatter(
    ###Explicitly using .index on x-axis###
    x=df_raw[df_raw[col_name]!=0].index,
    y=[df_raw['Close'][j] for j in df_raw[df_raw[col_name]!=0].index],
    mode='markers',
    marker=dict(
        size=8,
        color='black',
        symbol='cross'
    ),
    name='Cdl Pattern'
))

为了纠正这一点，必须将X轴转换为我使用pd.series的行号。如果您知道的话，请建议一种更好的方法。

fig = go.Figure()
fig.add_trace(go.Scatter(
    y=validation_df['Close'],
    mode='lines+markers',
    name='Original Plot'
))

fig.add_trace(go.Scatter(
    ###using pd.Series to get row numbers for x-axis###
    x=pd.Series(df_raw[col_name].to_list())[pd.Series(df_raw[col_name].to_list())!=0].index,

    y=[df_raw['Close'][j] for j in df_raw[df_raw[col_name]!=0].index],
    
    mode='markers',
    marker=dict(
        size=8,
        color='black',
        symbol='cross'
    ),
    name='Cdl Pattern'
))