Question

我有以下数据框，列中有某些值：ID'重复。我想按datafarme分组

 df = pd.DataFrame(
    {'Score': [0.65, 0.57, 0.56, 0.689, 0.56, 0.85, 0.45, 0.15, 0.14, 0.4678], 
    'ID': ['AAA', 'BBB', 'SAS', 'SAP', 'TCS', 'BBB', 'CTC', 'CTC', 'CTC', 'CTC'], 
    'Sample': ['AE01', 'AE01', 'AE03', 'AE03', 'AE03', 'AE05', 'AE05', 'AE05', 'AE05', 'AE05'], 
    'Freq': [1, 14, 14, 15, 16, 17, 18, 19, 20, 21], 
    'Type': ['Non', 'IND', 'IND', 'IND', 'IND', 'IND', 'IND', 'IND', 'IND', 'IND']},
    columns=['Sample','ID','Type','Score','Freq'])

当我尝试将“Re”列添加到group by选项时，通过使用transform

跟随单行将其分配到原始数据帧df

df['Re']=df.drop_duplicates(['Sample','ID']).groupby(['ID']).transform(size)

我收到了以下错误，

ValueError: Wrong number of items passed 4, placement implies 1

我知道我可以进行合并以获得按结果分配给组的原始数据帧，但是当我合并时，我得到的某些行在原始df中重复多次，这是我不需要的。任何更好的解决方案都会很棒。谢谢

Answer 1

我认为您需要先分配DataFrame的新drop_duplicates输出，因为显然drop_duplicates会删除一些重复行（如果存在），因此无法分配新列Re到原始df。

另一个问题是需要将Series添加到[] for groupby，因为transform仅适用于Series。如果使用size，您可以使用一些不用于分组的列，此处为ID。

df = df.drop_duplicates(['Sample','ID'])
df['Re']= df.groupby(['ID'])['Sample'].transform('size')
print (df)
  Sample   ID Type  Score  Freq  Re
0   AE01  AAA  Non  0.650     1   1
1   AE01  BBB  IND  0.570    14   2
2   AE03  SAS  IND  0.560    14   1
3   AE03  SAP  IND  0.689    15   1
4   AE03  TCS  IND  0.560    16   1
5   AE05  BBB  IND  0.850    17   2
6   AE05  CTC  IND  0.450    18   1

df = df.drop_duplicates(['Sample','ID'])
df['Re']= df.groupby(['ID'])['Type'].transform('size')
print (df)
  Sample   ID Type  Score  Freq  Re
0   AE01  AAA  Non  0.650     1   1
1   AE01  BBB  IND  0.570    14   2
2   AE03  SAS  IND  0.560    14   1
3   AE03  SAP  IND  0.689    15   1
4   AE03  TCS  IND  0.560    16   1
5   AE05  BBB  IND  0.850    17   2
6   AE05  CTC  IND  0.450    18   1

如果不指定，请获取NaN：

df['Re']= df.drop_duplicates(['Sample','ID']).groupby(['ID'])['Sample'].transform('size')
print (df)
  Sample   ID Type   Score  Freq   Re
0   AE01  AAA  Non  0.6500     1  1.0
1   AE01  BBB  IND  0.5700    14  2.0
2   AE03  SAS  IND  0.5600    14  1.0
3   AE03  SAP  IND  0.6890    15  1.0
4   AE03  TCS  IND  0.5600    16  1.0
5   AE05  BBB  IND  0.8500    17  2.0
6   AE05  CTC  IND  0.4500    18  1.0
7   AE05  CTC  IND  0.1500    19  NaN
8   AE05  CTC  IND  0.1400    20  NaN
9   AE05  CTC  IND  0.4678    21  NaN

数据帧上的pandas groupby和分配回orignal df失败的转换

1 个答案: