在同一图中分散多个数据框

时间:2018-11-13 16:50:01

标签: python pandas scatter-plot

我正在使用for循环在同一pd.plot.scatterplot上散布多个数据帧,但是每次循环返回时,它都会打印一个色条。 在周期结束时,如何只有一个颜色条?

这是我的代码

if colormap is None: colormap='jet'
f,ax = plt.subplots()
for i, data in enumerate(wells):
    data.plot.scatter(x,y, c=z, colormap=colormap, ax=ax)
ax.set_xlabel(x); ax.set_xlim(xlim)
ax.set_ylabel(y); ax.set_ylim(ylim)
ax.legend()
ax.grid()
ax.set_title(title)

1 个答案:

答案 0 :(得分:2)

这可以通过使用图形并将轴添加到同一子图中来实现:

import pandas as pd
import numpy as np

# created two dataframes with random values
df1 = pd.DataFrame(np.random.rand(25, 2), columns=['a', 'b'])
df2 = pd.DataFrame(np.random.rand(25, 2), columns=['a', 'b'])

然后:

fig = plt.figure()
for i, data in enumerate([df1, df2]):
    ax = fig.add_subplot(111)
    ax = data.plot.scatter(x='a', y='b', ax=ax,
                           c='#00FF00' if i == 0 else '#FF0000')

plt.show()

Resulting image with two dataframes plotted in one figure

您可以根据需要添加标签和其他元素。