python - 如何将数据框的值用作列并有选择地将值放入其中？

Question

我刚开始使用 pandas 数据框，但实际上遇到了一个问题。

我有一个主字典，它基本上是代码中的一个数据框，如下所示:

A      B      ATTR      VAL
AA     BB     type      circular
AA     BB     size      2mm
AA     BB     shape     square
AA     BB     tip       bend

我将提供一个字符串来检查这个数据框并选择单词并返回一个看起来非常像这样的输出:

A     B     type       size   shape   tip
AA    BB    circular   2mm    square  bend

我将针对大量数据执行此操作，但现在我正在对此进行尝试，完成后我将在该大型数据集上应用相同的代码。

main_dict = pd.read_excel('File.xlsx')
header = main_dict.iloc[:,0:2]
data = main_dict.iloc[:,2:4]
matr_desc = "AA BB  2mm circular bend".split()

for i,j in enumerate(main_dict['Attribute']):
  for k,l in enumerate(main_dict['Values']):
    print(l)
    header[j][k] = l

在此之后我陷入困境，我知道这可能很容易，但由于我是新手，所以我在构建它时遇到了一些困难。

Answer 1

您可以使用 unstack :

df.set_index(['A','B', 'ATTR']).unstack('ATTR')

或pivot_table :

df.pivot_table(index=['A', 'B'], columns='VAL', values='ATTR', aggfunc='first')

两者都会产生预期的输出。但是，请注意，跨越 A、B 和 ATTR 列的重复项(即这三列都具有相同值的多行)可能具有不需要的结果。