Python Pandas DataFrame melt方法用法及代码示例

例子

考虑以下 DataFrame ：

df = pd.DataFrame({"name":["alex","bob","cathy"], "age":[40,50,60], "height":[150,160,170]})
df



   name   age  height
0  alex   40   150
1  bob    50   160
2  cathy  60   170

基本用法

这里，标识符列是 name ，要取消透视的列是 age 和 height ：

df.melt(id_vars="name", value_vars=["age","height"])



   name   variable  value
0  alex   age       40
1  bob    age       50
2  cathy  age       60
3  alex   height    150
4  bob    height    160
5  cathy  height    170

在这里，我们实际上可以省略 value_vars 参数，因为默认情况下，除了标识符列 ( name ) 之外的所有列都将被取消透视。

指定value_vars

假设我们只想取消透视一列而不是两列。我们可以使用 value_vars 参数简单地指定要取消透视的列：

df.melt(id_vars="name", value_vars="age")



   name   variable  value
0  alex   age       40
1  bob    age       50
2  cathy  age       60

请注意 height 列是如何被剥离的。如果我们想保留高度列，我们必须将其包含在 id_vars 参数中，如下所示：

df.melt(id_vars=["name","height"], value_vars="age")



   name   height  variable  value
0  alex   150     age       40
1  bob    160     age       50
2  cathy  170     age       60

指定var_name和value_name

在这里，我们再次展示df供您参考：

df



   name   age  height
0  alex   40   150
1  bob    50   160
2  cathy  60   170

正如我们在上面的示例中看到的，新的列标签默认为 "variable" 和 "value" ，我们可以通过指定 var_name 和 value_name 来更改它，如下所示：

df.melt(id_vars="name", var_name="attribute", value_name="number")



   name   attribute  number
0  alex   age        40
1  bob    age        50
2  cathy  age        60
3  alex   height     150
4  bob    height     160
5  cathy  height     170