從 numpy arrays 創建 dataframe

Question

我有以下代碼，我被一些 numpy ZA3CBC3F9D0CE2F2C19CZE1B66714 創建一個 pandas dataframe。

def gradient_descent_classification_prediction(x_test):
    y_pred = pd.DataFrame()
    for i in range(0 , 10):
        print(i)
        b = pd.read_csv("theta"+str(i)+".csv" , delimiter=",")
        y = pred(x_test , b) #outputs are 2d-s numpy arrays (10000 , 1) 
        y_pred= y_pred.append(y.tolist()) 
    y_pred.to_csv("./y_pred.csv" , index=False)

我想在 dataframe 中有一個列，每個 y 由 function pred(x_test, b) 計算

我得到的 output 是一個單列 dataframe ，只有 ZC1C4145268E683845 的 output 的 ZC1C1145268E68384DZ。

你可以幫幫我嗎？

Answer 1

有很多方法可以做到這一點，但基本上 append 到 dataframe 的列表並不能確保其形狀。 將其保存為字典或列表更容易，然后在寫入之前轉換為數據框。 下面我使用字典：

def gradient_descent_classification_prediction(x_test):
    y_pred = {}
    for i in range(0 , 10):

        b = pd.read_csv("theta"+str(i)+".csv" , delimiter=",")
        y = pred(x_test , b) #outputs are 2d-s numpy arrays (10000 , 1) 
        y_pred["theta"+str(i)] = y
    
    pd.DataFrame(y_pred).to_csv("./y_pred.csv" , index=False)