绘制时间序列散点图

Question

我有以下数据 -

ProductName    01/01/2016    01/07/2016    01/14/2017
ABC              12             34            51
XYZ               9             76            12
PQR              12             23             7
DEF              54              4            34

我想绘制一个显示每天总销售额的时间序列散点图。我创建了以下函数 -

def scatterplot(x_data, y_data, x_label, y_label, title):
_, ax = plt.subplots()
ax.scatter(x_data, y_data, s = 30, color = '#539caf', alpha = 0.75)

ax.set_title(title)
ax.set_xlabel(x_label)
ax.set_ylabel(y_label)

我对如何调用这个函数来获得我想要的结果感到困惑。该图应在 x 轴上显示日期，在 y 轴上显示总销售额。

Answer 1

如果您的数据在 pandas DataFrame 中，您可以将 headers 列作为 x 值和沿垂直轴的数据总和（即售出的产品总数）天）作为 y 值。

import pandas as pd
import matplotlib.pyplot as plt

# replicate Data from question in DataFrame
v = [[12,34,51], [9,76,12], [12,23,7], [54,4,34]]
df = pd.DataFrame(v, columns=["01/01/2016","01/07/2016","01/14/2017"], 
                      index=["ABC", "XYZ", "PQR", "DEF"])
print(df)


def scatterplot(x_data, y_data, x_label, y_label, title):
    fig, ax = plt.subplots()
    ax.scatter(x_data, y_data, s = 30, color = '#539caf', alpha = 0.75)

    ax.set_title(title)
    ax.set_xlabel(x_label)
    ax.set_ylabel(y_label)
    fig.autofmt_xdate()

#use column headers as x values
x = pd.to_datetime(df.columns, format='%m/%d/%Y')
# sum all values from DataFrame along vertical axis
y = df.values.sum(axis=0)    
scatterplot(x,y, "x_label", "y_label", "title")

plt.show()

绘制时间序列散点图

Plot timeseries scatterplot

python

pandas

matplotlib

time-series

scatter-plot