Matplotlib:来自3列pandas数据框的pcolormesh或pcolor

时间:2018-08-30 22:45:49

标签: python pandas matplotlib

我有一个文件“ mydata.tmp”,其中包含3个这样的列:

3.81107 0.624698 0.000331622 
3.86505 0.624698 0.000131237 
3.91903 0.624698 5.15136e-05 
3.97301 0.624698 1.93627e-05 
1.32802 0.874721 1.59245 
1.382   0.874721 0 
1.43598 0.874721 0 
1.48996 0.874721 4.27933 

然后我要制作一个热图颜色图,其中前两列是坐标,第三列是该坐标的值。

此外,我想在对数刻度中设置第三列。

我该怎么做?

我使用散点图类型尝试了以下代码

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

df = pd.read_csv('mydata.tmp', delim_whitespace=True, 
                     comment='#',header=None,
                     names=['a','b','c'])

fig, ax = plt.subplots()

sc = ax.scatter(df.a, df.b, c=df.c,  cmap="GnBu", s=400, 
     norm=matplotlib.colors.LogNorm())

fig.colorbar(sc, ax=ax)

plt.show()

,我得到下面显示的图片(忽略x轴的比例)。但是,我想获得在使用此代码进行GNUplot时获得的结果(我还附加了GNUplot图像)

plot mydata.tmp using 1:2:3 with image

也许我必须使用pcolormesh? 谢谢!

GNUplot图片: GNUplot Image

Matplotlib图片: Matplotlib Image

当我尝试使用Khalil代码时,我得到以下图像: enter image description here

1 个答案:

答案 0 :(得分:1)

尝试一下。测试并处理我拥有的一些数据。 间距非常重要。根据您想要的绘图网格进行设置。间距越大,图像越平滑,但计算时间越长。

import pandas as pd
import matplotlib.pyplot as plt
import scipy.interpolate
import numpy as np
import matplotlib.colors as colors

# import data
df = pd.read_csv('mydata.tmp', delim_whitespace=True, 
                     comment='#',header=None,
                     names=['1','2','3'])

x = df['1']
y = df['2']
z = df['3']

# Set up a regular grid of interpolation points

spacing = 500
xi, yi = np.linspace(x.min(), x.max(), spacing), np.linspace(y.min(), 
                     y.max(), spacing)

XI, YI = np.meshgrid(xi, yi)

# Interpolate
rbf = scipy.interpolate.Rbf(x, y, z, function='linear')

ZI = rbf(XI, YI)

#plot

fig, ax = plt.subplots()

sc = ax.imshow(ZI, vmin=z.min(), vmax=z.max(), origin='lower',
            extent=[x.min(), x.max(), y.min(), 
                    y.max()], cmap="GnBu", norm=colors.LogNorm(vmin=ZI.min(),
                    vmax=ZI.max()))

fig.colorbar(sc, ax=ax, fraction=0.05, pad=0.01)

plt.show()