按列拆分 pandas dataframe 然后 append

Question

我有 dataframe：

我想保留前三列，然后將它們堆疊在接下來的 3 列之上，這樣我就有一個 dataframe 最后只有["userID", "tweetID", "time"] 。

Answer 1

df_top = tree_df[['userID_start', 'tweetID_start', 'time_start']]
df_top.columns = ['userID', 'tweetID', 'time']
df_bottom = tree_df[['userID_end', 'tweetID_end', 'time_end']]
df_bottom.columns = ['userID', 'tweetID', 'time']
final_df = pd.concat([df_top, df_bottom])

這行得通。

Answer 2

看來您需要pd.wide_to_long ：

df.columns=df.columns.str.split('_').map(lambda x : '_'.join(x[::-1]))
s=pd.wide_to_long(df.reset_index(),['end','start'],i='index',j='drop',sep='_',suffix='\w+').stack().unstack(-2)

Answer 3

start_cols = [c for c in df.columns if c.endswith('_start')]
end_cols = [c for c in df.columns if c.endswith('_end')]

def rename(x): return x.split('_')[0]

output_df = pd.concat([
    df[start_cols].rename(columns=rename), 
    df[end_cols].rename(columns=rename)
])

您可以將兩個數據幀中的開始列和結束列子集並將它們 pd.concat 組合在一起以堆疊。