Matplotlib:在两点之间绘制直线。。。但是以一种不寻常的方式

Question

我正在学习使用Matplotlib可视化数据，我想以非常定制的方式绘制一条线。我是Matplotlib新手，我不知道我想做的事情是否可能。下面是：

设置：假设你有一组随机的（x，y）点作为线图。为了便于说明，我将4个不同的线图压缩成一个图，但是想象一下，如果下面的4行中的每一行都被分解成各自的图。

[！[在此输入图像描述][1]][1]

步骤1）对于每个折线图A、B、C、D，转到全局最大值并将其称为“X”。

步骤4（可能吗？）画一条连接“X”和“Y”的线。可以在matplotlib中绘制这条线吗？非常感谢。

相关代码如下：

import pandas as pd
import numpy as np

df = pd.DataFrame(np.random.randn(50, 4), 
        index=pd.date_range('1/1/2000', periods=50), columns=list('ABCD'))
df = df.cumsum()
df.plot();

Answer 1

我认为你需要定义计算局部最大值的特定逻辑，而不仅仅是使用“时钟”比喻，但是一旦你用“代码”、“代码”熊猫或其他库定义了这个逻辑，你就可以像我一样创建一个数据帧。从那里，你应该能够从下面产生结果。

如果您可以使用seborn（构建在matplotlib之上），我认为这会更容易一点，因为您可以通过hue参数在一行中为每个类别创建所有行的代码。您需要使用要绘制的这些行创建一个新的数据帧。我通过对值进行排序并获得每个组的尾部值来做到这一点。见下面可重复的例子。

示例1（绘制局部最大值）：

import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
import matplotlib.dates as mdates
import seaborn as sns
fig, ax = plt.subplots()
plt.style.use('classic')
df = pd.DataFrame(np.random.randn(50, 4), 
        index=pd.date_range('1/1/2000', periods=50), columns=list('ABCD'))
df = df.cumsum()
df = df.melt(ignore_index=False).reset_index()
sns.lineplot(data=df, x="index", y="value", hue="variable", ax=ax)
lines_max = (df.sort_values('value').groupby("variable").tail(1)).sort_values('variable')
lines_local_max = df[((df['variable'] == lines_max['variable'].iloc[0]) & (df['index'] > lines_max['index'].iloc[0]))
                | ((df['variable'] == lines_max['variable'].iloc[1]) & (df['index'] > lines_max['index'].iloc[1]))
                | ((df['variable'] == lines_max['variable'].iloc[2]) & (df['index'] > lines_max['index'].iloc[2]))
                | ((df['variable'] == lines_max['variable'].iloc[3]) & (df['index'] > lines_max['index'].iloc[3]))]
lines_local_max = (lines_local_max.sort_values(['variable', 'value']).groupby("variable").tail(1))
lines = lines_max.append(lines_local_max).sort_values('variable')
lines
sns.lineplot(data=lines, x="index", y="value", hue="variable", marker="o",
             style='variable', dashes=[(2, 2), (2, 2), (2, 2), (2, 2)], legend=False, ax=ax)
x_dates = pd.to_datetime(df['index'].unique())
plt.xticks(x_dates[0::7], rotation=45, ha='center')
ax.xaxis.set_major_formatter(mdates.DateFormatter('%b-%d-%Y'))

例2（只是画一条线到最后，而不是定义局部最大值——目的只是向你展示如何从最大点到另一个定义点画一条线）：

import pandas as pd
import seaborn as sns
import matplotlib.pyplot as plt
fig, ax = plt.subplots()
plt.style.use('classic')
fmri = sns.load_dataset("fmri")
fmri = fmri.groupby(['event', 'timepoint'], as_index=False)['signal'].mean()
sns.lineplot(data=fmri, x="timepoint", y="signal", hue="event")
lines_max = (fmri.sort_values('signal').groupby("event").tail(1))
lines_last = (fmri.sort_values('timepoint').groupby("event").tail(1))
lines = lines_max.append(lines_last)
sns.lineplot(data=lines, x="timepoint", y="signal", hue="event", marker="o", style='event', dashes=[(2, 2), (2, 2)])

示例3（另一个示例使用您提供的数据绘制一条线，但不定义局部最大值-目的只是向您展示如何从最大值点到另一个定义点绘制一条线）：使用您提供的数据示例：

import pandas as pd
import numpy as np
fig, ax = plt.subplots()
df = pd.DataFrame(np.random.randn(50, 4), 
        index=pd.date_range('1/1/2000', periods=50), columns=list('ABCD'))
df = df.cumsum()
df = df.melt(ignore_index=False).reset_index()
sns.lineplot(data=df, x="index", y="value", hue="variable", ax=ax)
lines_max = (df.sort_values('value').groupby("variable").tail(1))
lines_last = (df.sort_values('index').groupby("variable").tail(1))
lines = lines_max.append(lines_last).sort_values('variable')
sns.lineplot(data=lines, x="index", y="value", hue="variable", marker="o",
             style='variable', dashes=[(2, 2), (2, 2), (2, 2), (2, 2)], legend=False, ax=ax)
x_dates = df['index'].dt.strftime('%Y-%m-%d').sort_values().unique()
ax.set_xticklabels(labels=x_dates, rotation=45, ha='center')