PROSAGA码农传奇-模型可视化-使用pandas进行多索引和绘图

使用pandas进行多索引和绘图

作者: Fire ming
发布时间: 2024-09-04 05:56:20 (6天前)
转自：

<div class =“post-text”itemprop =“text”>
  <P>
    删除索引级别，转置和绘图。
  </p>
   <pre>
    <code>
      df.reset_index('is_greedy', drop=True).T.plot()

</code>
  </pre>
  <H3>
    样本数据
  </H3>
   <pre>
    <code>
      import pandas as pd
import numpy as np

df = pd.DataFrame(np.random.randint(30, 50, (3, 11)), 
                  index=['no_transfer', 'transfer', 'random_transfer'])
df = df.set_index(10, append=True)
df.index.names=['config', 'is_greedy']
#                            0   1   2   3   4   5   6   7   8   9
#config          is_greedy                                        
#no_transfer     40         35  49  47  36  48  47  39  48  38  32
#transfer        33         35  33  45  38  38  45  36  30  46  36
#random_transfer 36         32  41  36  41  37  35  40  42  49  32

df.reset_index('is_greedy', drop=True).T.plot()

</code>
  </pre>
  <P>
    <a href="https://i.stack.imgur.com/eWQdV.png" rel="nofollow noreferrer">
      <img src =“https://i.stack.imgur.com/eWQdV.png”alt =“在此处输入图片说明”/>
    </A>
  </p>
  <HR />
  <P>
    或者遍历行：
  </p>
   <pre>
    <code>
      import matplotlib.pyplot as plt

fig, ax = plt.subplots()
for idx, row in df.iterrows():
    ax.plot(df.columns, row, label=idx[0])
plt.legend()
plt.show()

</code>
  </pre>
</DIV>