一个图matplotlib中的两个DataFrame图

Question

我想在一个图中绘制两个 DataFrame。不过，我看过类似的帖子，但似乎都没有。

我的数据框的前 5 行如下所示：

df1

    name    type        start       stop    strand
0   geneA   transcript  2000        7764        +
1   geneA   exon        2700        5100        +
2   geneA   exon        6000        6800        +
3   geneB   transcript  9000        12720       -
4   geneB   exon        9900        10100       -

df2

    P1      P2       P3      P4
0   0.28    0.14    0.19    0.19
1   0.30    0.16    0.17    0.20
2   0.26    0.13    0.20    0.12
3   0.21    0.13    0.25    0.15
4   0.31    0.03    0.24    0.20

我希望情节看起来像这样：

我试过这样做：

import pandas as pd
import matplotlib.pyplot as plt
import numpy as np

ax = df1.plot()

df1.plot(ax=ax)

但是，输出没有意义。 我将感谢有关如何实现这一目标的建议/解决方案。

Answer 1

您可以使用subplots 这样做（因为很难理解应该如何绘制两个df，所以我提供了一个通用示例）

Import matplotlib.pyplot as plt

fig,axes = plt.subplots(4,1) #4 rows, one column
for ax in axes:
   plt.plot(X1,y1 ax =ax) # loop over each subplot and create a plot
   plt.plot(X2,y2, ax = ax)

Answer 2

这是一个最小的例子：

import matplotlib.pyplot as plt

f, axes = plt.subplots(nrows=len(df2.columns)+1, sharex=True, )

# plots for df2 columns
i = 0
for col in df2.columns:
    df2[col].plot(ax=axes[i])
    axes[i].set_ylim(0, 1.2)
    axes[i].set_ylabel(col)
    i+=1
    
## code to plot annotations
# axes[-1].plot(…)
axes[-1].set_xlabel('Genomic position')

# remove space between plots
plt.subplots_adjust(hspace=0)

这是完整的图表：

f, axes = plt.subplots(nrows=len(df2.columns)+1, sharex=True, )

# plots for df2 columns
i = 0
for col in df2.columns:
    df2[col].plot(ax=axes[i], color='#505050')
    axes[i].set_ylim(0, 1.3)
    axes[i].set_ylabel(col)
    i+=1
    
## code to plot annotations
axes[-1].set_xlabel('Genomic position')
axes[-1].set_ylabel('annotations')
axes[-1].set_ylim(-0.5, 1.5)
axes[-1].set_yticks([0, 1])
axes[-1].set_yticklabels(['−', '+'])

for _, r in df1.iterrows():
    marker = '|'
    lw=1
    if r['type'] == 'exon':
        marker=None
        lw=8
    y = 1 if r['strand'] == '+' else 0
    axes[-1].plot((r['start'], r['stop']), (y, y),
                  marker=marker, lw=lw,
                  solid_capstyle='butt',
                  color='#505050')
    
# remove space between plots
plt.subplots_adjust(hspace=0)