简体   繁体   English

plot 从 pandas dataframe 具有负值和正值

[英]plot from pandas dataframe with negative and positive values

I have a dataframe which looks like this:我有一个 dataframe 看起来像这样:

MM Initial Energy   MM Initial Angle    QM Energy   QM Angle
0   13.029277   120.0   18.048  120.0
1   11.173115   125.0   15.250  125.0
2   9.411475    130.0   12.668  130.0
3   7.762888    135.0   10.309  135.0
4   6.239025    140.0   8.180   140.0
5   4.853004    145.0   6.286   145.0
6   3.617394    150.0   4.633   150.0
7   2.544760    155.0   3.226   155.0
8   1.646335    160.0   2.070   160.0
9   0.934298    165.0   1.166   165.0
10  0.419003    170.0   0.519   170.0
11  0.105913    175.0   0.130   175.0
12  0.000000    -180.0  0.000   -180.0
13  0.105988    -175.0  0.130   -175.0
14  0.420029    -170.0  0.519   -170.0
15  0.937312    -165.0  1.166   -165.0
16  1.650080    -160.0  2.070   -160.0
17  2.548463    -155.0  3.227   -155.0
18  3.621227    -150.0  4.633   -150.0
19  4.856266    -145.0  6.286   -145.0
20  6.236939    -140.0  8.180   -140.0
21  7.760035    -135.0  10.309  -135.0
22  9.409117    -130.0  12.669  -130.0
23  11.170671   -125.0  15.251  -125.0
24  13.033293   -120.0  18.048  -120.0

I want to plot the data with Angles on the x-axis and energy on the y.我想 plot X 轴上的角度和 y 上的能量数据。 This sounds fairly simple, however what happens is that pandas or matplotlib sorts the X-axis values in a such a manner that my plot looks split.这听起来很简单,但是发生的情况是 pandas 或 matplotlib 以这样一种方式对 X 轴值进行排序,使我的 plot 看起来分裂。 This is what it looks like:这是它的样子:

原来的

However, this is how I want it:但是,这就是我想要的:

通缉情节 My code is as follows:我的代码如下:

df=pd.read_fwf('scan_c1c2c3h31_orig.txt', header=None, prefix='X')

df.rename(columns={'X0':'MM Initial Energy',
                          'X1':'MM Initial Angle',
                          'X2':'QM Energy', 'X3':'QM Angle'}, 
                 inplace=True)

df=df.sort_values(by=['MM Initial Angle'], axis=0, ascending=True)
df=df.reset_index(drop=False)

df2=pd.read_fwf('scan_c1c2c3h31.txt', header=None, prefix='X')
df2.rename(columns={'X0':'MM Energy',
                          'X1':'MM Angle',
                          'X2':'QM Energy', 'X3':'QM Angle'}, 
                 inplace=True)
df2=df2.sort_values(by=['MM Angle'], axis=0, ascending=True)
df2=df2.reset_index(drop=False)
df
df2
ax = plt.axes()

df.plot(y="MM Initial Energy", x="MM Initial Angle", color='red', linestyle='dashed',linewidth=2.0, ax=ax, fontsize=20, legend=True)

df2.plot(y="MM Energy", x="MM Angle", color='red', ax=ax, linewidth=2.0, fontsize=20, legend=True)

df2.plot(y="QM Energy", x="QM Angle", color='blue', ax=ax, linewidth=2.0, fontsize=20, legend=True)

plt.ylim(-0.05, 6)

ax.xaxis.set_major_locator(MultipleLocator(20))
ax.xaxis.set_minor_locator(MultipleLocator(10))
ax.yaxis.set_minor_locator(MultipleLocator(0.5))

plt.xlabel('Angles (Degrees)', fontsize=25)
plt.ylabel('Energy (kcal/mol)', fontsize=25)

What I am doing is, sorting the dataframe by 'MM Angles'/'MM Initial Angles' to avoid plot "scarambling" due to repeating values in the y-axis.The angles vary from -180 to 180, where I want the -180 and +180 next to each other.我正在做的是,将 dataframe 按“MM 角度”/“MM 初始角度”排序,以避免 plot 由于 y 轴上的重复值而“混乱”。角度从 -180 到 180 不等,我想要 - 180 和 +180 并排。

I have tried sorting the negative values in ascending order and positive values in descending order as suggested in this post , but I still get the same plot where x axis ranges from -180 to +180.我已尝试按照本文中的建议按升序对负值进行排序,并按降序对正值进行排序,但我仍然得到相同的 plot,其中 x 轴的范围从 -180 到 +180。 I have also tried matplotlib axis spines to recenter the plot, and I have also tried inverting the x-axis as suggested in this post , but still get the same plot.我还尝试了 matplotlib 轴以使 plot 居中,并且我还尝试按照本文中的建议反转 x 轴,但仍然得到相同的 Z32FA6E1B78A9D40248953E60564AC。 Additionally, I have also tried suggestion in this another post .此外,我还尝试过在另一篇文章中提出建议。 Any help will be appreciated.任何帮助将不胜感激。

If you don't need to rescale the plot, I would plot against the positive angles 0-360 and manually re-label the ticks:如果您不需要重新调整 plot,我将 plot 针对正角0-360并手动重新标记刻度:

fig, ax = plt.subplots()
(df.assign(Angle=df['MM Initial Angle']%360)
   .plot(x='Angle', y=['QM Energy','MM Initial Energy'], ax=ax)
)

ax.xaxis.set_major_locator(MultipleLocator(20))

x_ticks = ax.get_xticks()
x_ticks = [t-360 if t>180 else t for t in x_ticks]
ax.set_xticklabels(x_ticks)
plt.plot()

Output: Output: 在此处输入图像描述

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM