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