我有一个文件“ 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? 谢谢!
答案 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()