Python的大熊貓GROUPBY /追加列

Question

這是我的示例數據幀：

Index Param1 Param2 
    A  1  2 
    A  3  4 
    B  1  3 
    B  4  Nan 
    C  2  4 

我想獲得的是：

Index Param1 Param2 Param3 Param4 
    A  1  2  3  4 
    B  1  3  4 
    C  2  4 

什麼是實現這一目標的最佳方式使用熊貓嗎？ 在此先感謝您的幫助。

Answer 1

您可以使用groupby與unstack：

def f(x): 
    return (pd.DataFrame(np.sort(x.values.ravel()))) 

df = df.groupby('Index')['Param1','Param2'].apply(f).unstack() 
df.columns = df.columns.droplevel(0) 
print (df) 
     0 1  2  3 
Index     
A  1 2  3  4 
B  1 3  4 Nan 
C  2 4 None None 

，因爲如果使用Series得到：

TypeError: Series.name must be a hashable type

與cumcount另一種解決方案：

df = df.set_index('Index').stack().reset_index(name='vals') 
df['g'] = 'Param' + df.groupby('Index').cumcount().add(1).astype(str) 
df = df.pivot(index='Index', columns='g', values='vals') 
print (df) 
g  Param1 Param2 Param3 Param4 
Index         
A   1.0  2.0  3.0  4.0 
B   1.0  3.0  4.0  NaN 
C   2.0  4.0  NaN  NaN 

Answer 2

import numpy as np 
import pandas as pd 

df = pd.DataFrame({'Index': ['A', 'A', 'B', 'B', 'C'], 'Param1': [1, 3, 1, 4, 2], 
        'Param2': [2, 4, 3, np.nan, 4]}).set_index('Index') 
print(df) 

#  Param1 Param2 
# Index     
# A   1  2.0 
# A   3  4.0 
# B   1  3.0 
# B   4  NaN 
# C   2  4.0 

def fn(g): 
    return pd.Series(g.values.ravel()) 

res = df.groupby(df.index).apply(fn).unstack() 
res.columns = ['Param1', 'Param2', 'Param3', 'Param4'] 
print(res) 

#  Param1 Param2 Param3 Param4 
# Index         
# A   1.0  2.0  3.0  4.0 
# B   1.0  3.0  4.0  NaN 
# C   2.0  4.0  NaN  NaN