Создание списка элементов из dataframe по последней букве в строке

Question

Извините за то, что кажется невероятно основной проблемой, но на самом деле я не мог найти ответы на это. У меня есть прямая функция добавления элементов (символов запаса) в список на основе их последней буквы. Для начала я использую список. «nyse» - это dataframe. Функция такова:

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)