通過字符串中的最後一個字母創建數據框中的項目列表

Question

對於什麼感覺像一個令人難以置信的基本問題抱歉，但我實際上找不到任何答案。我有一個簡單的功能，可以根據項目的最後一個字母添加項目（股票代碼）。首先，我使用一個列表。 'nyse'是數據框。功能是這樣的：

nysewarrants =() 

def warrants(x, y): 
    for e in x: 
     if x.Symbol[e][-1] == 'W': 
      y.append(e) 
    return y 

warrants(nyse, nysewarrants) 

當前錯誤是「KeyError：'Symbol'」。我嘗試了「print（nyse.Symbol [9] [ - 1]）」）並獲得了預期的輸出。

非常感謝您的幫助。

Answer 1

對pandas進行迭代按DataFrame逐列迭代。由於要通過行迭代，你需要使用iterrows()

nysewarrants = [] 

def warrants(x, y): 
    for index, row in x.iterrows(): 
     if x.Symbol[index][-1] == 'W': 
      y.append(row) 
    return y 

warrants(nyse, nysewarrants)