Python - 如何使用具有相应值的现有列中的唯一值在数据框中创建新列?

时间:2015-02-10 16:40:25

标签: python pandas

小代码是......

import pandas as pd

#INPUT FILE INFORMATION 
path = 'C:\Users\BDomitz\Desktop\Python\Stack_Example.xlsx'
sheet = "Sheet1"

#READ FILE
dataframe = pd.io.excel.read_excel(path, sheet)

我当前数据帧的输出......

   date       animals       quantity
0  2015-02-10    dogs       1
1  2015-02-11    cats       2
2  2015-02-11    pigs       5

我希望它看起来像......

   date       animals       quantity    dogs   cats    pigs
0  2015-02-10    dogs       1            1      0        0
1  2015-02-11  cats, pigs   2            0      2        5

我很感激帮助。

1 个答案:

答案 0 :(得分:2)

从您的数据框开始:

In [9]: df
Out[9]:
         date animals  quantity
0  2015-02-10    dogs         1
1  2015-02-11    cats         2
2  2015-02-11    pigs         5

您可以使用pivot方法指定应将哪些列用作索引,列名称以及值:

In [10]: df.pivot(index='date', columns='animals', values='quantity').fillna(0)
Out[10]:
animals     cats  dogs  pigs
date
2015-02-10     0     1     0
2015-02-11     2     0     5

除了“动物”之外,它还能为您提供所需的输出。和'数量'列。他们需要在那里吗?