Python DataFrame 将行转换为列

Question

我有一个 csv，用于存储有关特定对象和日期的信息。

Device  Date        Category    Amount
Pen     01/01/2014  A           12
Pen     01/01/2014  B           42
Pen     01/01/2014  C           10
Pen     01/01/2014  D           5
Pen     02/01/2014  A           7
Pen     02/01/2014  B           52
Pen     02/01/2014  C           1
Pen     02/01/2014  D           7
Pencil  01/01/2014  A           22
Pencil  01/01/2014  B           42
Pencil  01/01/2014  C           70
Pencil  01/01/2014  D           8

我想将其读入 DataFrame 并将 Category 设为一列，将 Amounts 设为特定设备，并将日期设为一行。这将使数据集更小。

Device  Date        A   B   C   D
Pen    01/01/2014   12  42  10  5
Pen    02/01/2014   7   52  1   7
Pencil  01/01/2014  22  42  70  8

Answer 1

您可以使用pivot_table，其中要保留的列设置为index，转到标题的列设置为columns，填充输出数据框中单元格的列设置为@ 987654324@:

df.pivot_table(index=['Device', 'Date'], columns='Category', values='Amount').reset_index()

# Category  Device        Date     A     B   C  D
#        0     Pen  01/01/2014    12    42  10  5
#        1     Pen  02/01/2014     7    52   1  7
#        2  Pencil  01/01/2014    22    42  70  8

Answer 2

使用groupby 和unstack

df.groupby(['Device', 'Date', 'Category'])['Amount'].sum().unstack().reset_index()