Numpy vstack：具有1行的數據幀的關鍵錯誤0

Question

我的軟件99％的時間都能正常工作，但是我發現一個錯誤，如果pandas數據框只有一行，則會導致程序崩潰

def prep(df):        
    X = np.vstack(np.asarray((df['col1'],df['col2']),order='C',dtype='float')).T
    return df,X

現在當調用函數時出現此錯誤

KeyError: 0

我不確定發生了什么，但是注意到只有在df有一行時，它才會出現此錯誤。 大於1行的任何事物都像是魅力一樣...

Answer 1

我很久以前就寫了這個函數。 我現在學得更好的python

這是未來人們的解決方案

def prep(df):
    X = df[['col1','col2']].as_matrix()
    return df,X