使用 pd.melt 时如何保留索引并合并为 Seaborn 和 matplotlib 创建 DataFrame

Question

我正在尝试使用结构完全相同的两个相同的数据帧（预测的和观察到的）绘制子图...第一列是索引

下面的代码在使用 pd.melt 连接并合并时创建新索引如图所示，橙色线的索引从 1-5 更改为 6-10

我想知道是否有人可以修复下面的代码以保持橙色线的相同索引：

import pandas as pd 
import seaborn as sns
import matplotlib.pyplot as plt
%matplotlib inline

actual = pd.DataFrame({'a': [5, 8, 9, 6, 7, 2],
                       'b': [89, 22, 44, 6, 44, 1]})
predicted = pd.DataFrame({'a': [7, 2, 13, 18, 20, 2],
                       'b': [9, 20, 4, 16, 40, 11]})

# Creating a tidy-dataframe to input under seaborn
merged = pd.concat([pd.melt(actual), pd.melt(predicted)]).reset_index()
merged['category'] = ''
merged.loc[:len(actual)*2,'category'] = 'actual'
merged.loc[len(actual)*2:,'category'] = 'predicted'

g = sns.FacetGrid(merged, col="category", hue="variable")
g.map(plt.plot, "index", "value", alpha=.7)
g.add_legend();

Answer 1

由于您使用 melt 的方式，橙色行 ('variable' == 'b') 没有索引 0-5。如果您查看 pd.melt(actual)，索引与您的预期不符，IIUC。

以下是我将如何重新排列数据框：

merged = pd.concat([actual, predicted], keys=['actual', 'predicted'])
merged.index.names = ['category', 'index']
merged = merged.reset_index()
merged = pd.melt(merged, id_vars=['category', 'index'], value_vars=['a', 'b'])

Answer 2

将 ignore_index 变量设置为 false 以保留索引。例如

df = df.melt(var_name=‘species’, value_name=‘height’, ignore_index = False)

使用 pd.melt 时如何保留索引并合并为 Seaborn 和 matplotlib 创建 DataFrame

How to keep the index when using pd.melt and merge to create a DataFrame for Seaborn and matplotlib

python

pandas

matplotlib

seaborn

facet-grid