给定时间表示中的信号图,如何绘制标记相应时间索引的线?
具体来说,给定一个时间索引范围从 0 到 2.6(秒)的信号图,我想为列表[0.22058956, 0.33088437, 2.20589566]
绘制垂直的红线,表示相应的时间索引。
添加覆盖整个绘图窗口的垂直线而无需指定其实际高度的标准方法是plt.axvline
import matplotlib.pyplot as plt
plt.axvline(x=0.22058956)
plt.axvline(x=0.33088437)
plt.axvline(x=2.20589566)
或
xcoords = [0.22058956, 0.33088437, 2.20589566]
for xc in xcoords:
plt.axvline(x=xc)
您可以使用许多可用于其他绘图命令的关键字(例如color
,linestyle
,linewidth
...)。您可以传入关键字参数ymin
和ymax
如果您喜欢在轴上的 corrdinate(例如ymin=0.25
,ymax=0.75
将覆盖

matplotlib.pyplot.vlines
vs.matplotlib.pyplot.axvline
这些方法适用于使用 seaborn 和pandas.DataFrame.plot
生成的图,它们都使用matplotlib
。
区别在于vlines
接受x
的一个或多个位置,而axvline
允许一个位置。
单个位置:x=37
。
多个位置:x=[37, 38, 39]
。
vlines
将ymin
和ymax
作为 y 轴上的位置,而axvline
将ymin
和ymax
作为 y 轴范围的百分比。
将多行传递给vlines
时,将list
传递给ymin
和ymax
。
对于面向对象的 API,还有matplotlib.axes.Axes.vlines
和matplotlib.axes.Axes.axvline
。
如果您使用fig, ax = plt.subplots()
绘制图形,则将plt.vlines
或plt.axvline
分别替换为ax.vlines
或ax.axvline
。
import numpy as np
import matplotlib.pyplot as plt
xs = np.linspace(1, 21, 200)
plt.figure(figsize=(10, 7))
# only one line may be specified; full height
plt.axvline(x=36, color='b', label='axvline - full height')
# only one line may be specified; ymin & ymax specified as a percentage of y-range
plt.axvline(x=36.25, ymin=0.05, ymax=0.95, color='b', label='axvline - % of full height')
# multiple lines all full height
plt.vlines(x=[37, 37.25, 37.5], ymin=0, ymax=len(xs), colors='purple', ls='--', lw=2, label='vline_multiple - full height')
# multiple lines with varying ymin and ymax
plt.vlines(x=[38, 38.25, 38.5], ymin=[0, 25, 75], ymax=[200, 175, 150], colors='teal', ls='--', lw=2, label='vline_multiple - partial height')
# single vline with full ymin and ymax
plt.vlines(x=39, ymin=0, ymax=len(xs), colors='green', ls=':', lw=2, label='vline_single - full height')
# single vline with specific ymin and ymax
plt.vlines(x=39.25, ymin=25, ymax=150, colors='green', ls=':', lw=2, label='vline_single - partial height')
# place the legend outside
plt.legend(bbox_to_anchor=(1.0, 1), loc='upper left')
plt.show()
Seaborn 轴级图
import seaborn as sns
# sample data
fmri = sns.load_dataset("fmri")
# x index for max y values for stim and cue
c_max, s_max = fmri.pivot_table(index='timepoint', columns='event', values='signal', aggfunc='mean').idxmax()
# plot
g = sns.lineplot(data=fmri, x="timepoint", y="signal", hue="event")
# y min and max
ymin, ymax = g.get_ylim()
# vertical lines
g.vlines(x=[c_max, s_max], ymin=ymin, ymax=ymax, colors=['tab:orange', 'tab:blue'], ls='--', lw=2)
Seaborn 图级图
每个轴都必须迭代。
import seaborn as sns
# sample data
fmri = sns.load_dataset("fmri")
# used to get the index values (x) for max y for each event in each region
fpt = fmri.pivot_table(index=['region', 'timepoint'], columns='event', values='signal', aggfunc='mean')
# plot
g = sns.relplot(data=fmri, x="timepoint", y="signal", col="region", hue="event", kind="line")
# iterate through the axes
for ax in g.axes.flat:
# get y min and max
ymin, ymax = ax.get_ylim()
# extract the region from the title for use in selecting the index of fpt
region = ax.get_title().split(' = ')[1]
# get x values for max event
c_max, s_max = fpt.loc[region].idxmax()
# add vertical lines
ax.vlines(x=[c_max, s_max], ymin=ymin, ymax=ymax, colors=['tab:orange', 'tab:blue'], ls='--', lw=2, alpha=0.5)
对于'region = frontal'
,两个事件的最大值都出现在5
。
柱状图和直方图
请注意,条形图刻度位置具有zero-based索引,而与轴刻度标签无关,因此请根据条形图索引而不是刻度标签选择x
。
ax.get_xticklabels()
将显示位置和标签。
import pandas as pd
import seaborn as sns
# load data
tips = sns.load_dataset('tips')
# histogram
ax = tips.plot(kind='hist', y='total_bill', bins=30, ec='k', title='Histogram with Vertical Line')
_ = ax.vlines(x=16.5, ymin=0, ymax=30, colors='r')
# barplot
ax = tips.loc[5:25, ['total_bill', 'tip']].plot(kind='bar', figsize=(15, 4), title='Barplot with Vertical Lines', rot=0)
_ = ax.vlines(x=[0, 17], ymin=0, ymax=45, colors='r')
时间序列轴
要作为 x 轴的数据框中的日期必须是datetime dtype
。如果列或索引的类型不正确,则必须使用pd.to_datetime
进行转换。
如果使用日期数组或列表,请分别参考Converting numpy array of strings to datetime或Convert datetime list into date python。
x
将接受像'2020-09-24'
或datetime(2020, 9, 2)
这样的日期。
import pandas_datareader as web # conda or pip install this; not part of pandas
import pandas as pd
import matplotlib.pyplot as plt
from datetime import datetime
# get test data; this data is downloaded with the Date column in the index as a datetime dtype
df = web.DataReader('^gspc', data_source='yahoo', start='2020-09-01', end='2020-09-28').iloc[:, :2]
# display(df.head(2))
High Low
Date
2020-09-01 3528.030029 3494.600098
2020-09-02 3588.110107 3535.229980
# plot dataframe; the index is a datetime index
ax = df.plot(figsize=(9, 6), title='S&P 500', ylabel='Price')
# add vertical lines
ax.vlines(x=[datetime(2020, 9, 2), '2020-09-24'], ymin=3200, ymax=3600, color='r', label='test lines')
ax.legend(bbox_to_anchor=(1, 1), loc='upper left')
plt.show()

对于多行
xposition = [0.3, 0.4, 0.45]
for xc in xposition:
plt.axvline(x=xc, color='k', linestyle='--')
本站系公益性非盈利分享网址,本文来自用户投稿,不代表码文网立场,如若转载,请注明出处
评论列表(30条)