按字符串中的最后一个字母从数据框创建项目列表

时间:2017-02-21 03:08:02

标签: python python-3.x jupyter-notebook

对于一个令人难以置信的基本问题感到抱歉,但我实际上找不到任何答案。我有一个简单的功能,可以根据最后一个字母将项目(股票代码)添加到列表中。首先,我要使用列表。 ' 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:'符号'"。我试过" print(nyse.Symbol [9] [ - 1])"并获得了预期的产出。

非常感谢您的帮助。

1 个答案:

答案 0 :(得分: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)