在python熊貓中保留函數並創建一個新表格作爲SAS

Question

最近，我正在將SAS轉換爲Python熊貓。我有一個問題，大熊貓在SAS中是否具有保留功能。

我的SAS代碼：

data df1; 
retain col3 " "; 
set df; 
by ID ; 
if first.ID then col3=col1; 
else col3=col3; 

其他條件我已經是在SAS代碼：

data df1; 
retain col3; 
set df; 
by ID ; 
if first.ID then col3=1; 
else col3=col3+1; 

如下所示 對此我試圖轉換

下一頁SAS代碼：

proc sql; 
    create table t1 as 
    select 
     c1, c2, c3, c4, c5, flag, max(flag) as MAX_flag 
    from t1 
    group by c1, c2, c3, c5; 
    run; 

我在熊貓身上試過它，但看起來我犯了一些愚蠢的錯誤。如果有人知道如何複製sas代碼塊3到熊貓

t1=t1[['c1','c2','c3','c4','c5','c6']] 
    t1.loc[:,'Max_flag']=t1['flag'].max() 
    t1.groupby(['c1','c2','c3','c5']) 

在我的例如。 col3是B，col1是a。 在其上它應該做的條件是 df.groupby（[ 'ID']，as_index =假）。首先（）

我有2列ID，A. 我的要求是輸入數據DF1：

ID A 
    1 a 
    1 b 
    2 c 
    1 p 
    2 q 

輸出數據幀應該有一個列名稱爲B.它將按ID.first（）進行分組。並將col A的數據複製到col B以獲得所有分組的ID。

輸出應該DF1

ID A B 
    1 a a 
    1 b a 
    2 c c 
    1 p a 
    2 q c 

**My key requirement is to convert above SAS code to Pnadas** 

Answer 1

它看起來像（形成你已經試過的），你想擁有「COL4」等於「COL2」如果「COL1」不爲空，否則「COL4」應該是空白的。所以你不需要分組。 這裏是你怎麼做的情況下，我在這裏所說的情況：

In [80]: df = pd.DataFrame([[1,'a','b','c'],[2,'e','r','g'], [3,'BLANK', '', '']], columns=['ID','col1', 'col2','col3']) 

In [81]: df 
Out[81]: 
    ID col1 col2 col3 
0 1  a b c 
1 2  e r g 
2 3 BLANK   

In [82]: df['col4'] = np.where(df['col1'] == 'BLANK', 'BLANK', df['col2']) 

In [83]: df 
Out[83]: 
    ID col1 col2 col3 col4 
0 1  a b c  b 
1 2  e r g  r 
2 3 BLANK   BLANK 

Answer 2

考慮一個merge與groupby.first()：

df = df.merge(df.groupby('ID').first().reset_index(), on='ID').rename(columns={'A_x':'A', 'A_y':'B'}) 

# ID A B 
# 0 1 a a 
# 1 1 b a 
# 2 1 p a 
# 3 2 c c 
# 4 2 q c 

這也可以用groupby.nth()這是從零開始的推廣。下面需要在每個ID的第二個值：

df = df.merge(df.groupby('ID').nth(1).reset_index(), on='ID').rename(columns={'A_x':'A', 'A_y':'B'}) 

# ID A B 
# 0 1 a b 
# 1 1 b b 
# 2 1 p b 
# 3 2 c q 
# 4 2 q q 

而且還有groupby.last()：

df = df.merge(df.groupby('ID').last().reset_index(), on='ID').rename(columns={'A_x':'A', 'A_y':'B'}) 

# ID A B 
# 0 1 a p 
# 1 1 b p 
# 2 1 p p 
# 3 2 c q 
# 4 2 q q