python - Pandas df 操作 - 将数据分成 2 列

Question

我有一个示例数据框，其中包含 2 个指示列(下面的示例，产品和版本 - 其中版本号只能是 1 或 2)和一个包含实际数据的第三列。

product       version       data    
  a              1          8000
  a              2          1000
  b              1          4000
  b              2          2000
  c              1          9000
  c              2          1000
  d              1          2500
  d              2          3000

我正在尝试操作我的 df，以便我有 2 个数据列(比如 data1 和 data2)，它们对应于该产品的版本(从而使版本列变得多余)。见下文:

product       data1       data2    
  a           8000        1000
  b           4000        2000
  c           9000        1000
  d           2500        3000

有没有人能建议一个简单的解决方案来实现这一点？

Answer 1

您可以使用 pandas 函数 .pivot

旋转表格

>>> result = df.pivot(index='product', columns='version', values='data').reset_index()
>>> result
version product     1     2
0             a  8000  1000
1             b  4000  2000
2             c  9000  1000
3             d  2500  3000

>>> result.index.name = None
>>> result.columns = ['product', 'data1', 'data2']
>>> result
  product  data1  data2
0       a   8000   1000
1       b   4000   2000
2       c   9000   1000
3       d   2500   3000