Python Pandas 索引改革

Question

我有一个 CSV 表，其中包含类似于以下的数据:

Customer    Product    Quantity

C1          P1         2
C2          P1         3
C1          P2         4
C3          P2         1
C3          P1         0
C2          P2         5

我想把它转换成下面的格式:

Customer    P1    P2

C1          2     4
C2          3     5
C3          1     0

我知道我可以使用 pandas 导入文件，如下所示:

import pandas as pd
data = pd.read_csv('my_file.csv')

但是，我不知道之后该怎么办。

我曾尝试使用 pandas 多索引，但不太理解。如何重建新的 CSV 数据对象？任何帮助将不胜感激，谢谢。

附言我可以考虑多个指数来完成我的问题吗？例如，假设我们在表格的左侧添加了一列，其中包含购买日期。如何处理？

Answer 1

这是一个 pivot 的例子操作:

>>> df
  Customer Product  Quantity
0       C1      P1         2
1       C2      P1         3
2       C1      P2         4
3       C3      P2         1
4       C3      P1         0
5       C2      P2         5
>>> df.pivot(index="Customer", columns="Product", values="Quantity")
Product   P1  P2
Customer        
C1         2   4
C2         3   5
C3         0   1

reshaping上的文档部分推荐阅读。

(PS:我假设你的文件实际上有逗号，即使你没有显示任何 - 否则 pd.read_csv 将需要一个额外的参数，比如 delim_whitespace=True 或其他东西。)