簡體   English   中英

plot 從 pandas dataframe 具有負值和正值

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

我有一個 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

我想 plot X 軸上的角度和 y 上的能量數據。 這聽起來很簡單,但是發生的情況是 pandas 或 matplotlib 以這樣一種方式對 X 軸值進行排序,使我的 plot 看起來分裂。 這是它的樣子:

原來的

但是,這就是我想要的:

通緝情節 我的代碼如下:

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)

我正在做的是,將 dataframe 按“MM 角度”/“MM 初始角度”排序,以避免 plot 由於 y 軸上的重復值而“混亂”。角度從 -180 到 180 不等,我想要 - 180 和 +180 並排。

我已嘗試按照本文中的建議按升序對負值進行排序,並按降序對正值進行排序,但我仍然得到相同的 plot,其中 x 軸的范圍從 -180 到 +180。 我還嘗試了 matplotlib 軸以使 plot 居中,並且我還嘗試按照本文中的建議反轉 x 軸,但仍然得到相同的 Z32FA6E1B78A9D40248953E60564AC。 此外,我還嘗試過在另一篇文章中提出建議。 任何幫助將不勝感激。

如果您不需要重新調整 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: 在此處輸入圖像描述

暫無
暫無

聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM