在图例中使用pandas.DataFrame.plot的temprary标签

Question

我有一个pandas.DataFrame对象，其列名非常难看：

df = pd.DataFrame({'ugly name1':[1,2,3], 'ugly name2':[2,3,4]})

我想在图例中使用漂亮的数据名称绘制该数据框。我知道如何重命名专栏，但希望在绘图后保留这个丑陋的名字。

我的问题是下面相同过程中最简单的方法。

df = pd.DataFrame({'ugly name1':[1,2,3], 'ugly name2':[2,3,4]})
column_names = df.columns
df.columns = ['pretty1', 'pretty2']
df.plot()
df.columns = column_names
print df.columns #=> should print ['ugly name1', 'ugly name2']

Answer 1

绘制绘图后，您可以手动更改图例：

import matplotlib.pyplot as plt
import pandas as pd
df = pd.DataFrame({'ugly name1':[1,2,3], 'ugly name2':[2,3,4]})

df.plot()
plt.legend(['pretty1', 'pretty2'],loc='upper left')