Python Pandas Dataframe循环迭代

时间:2018-05-03 03:27:07

标签: python pandas for-loop iteration

import pandas as pd
import re

df1 = pd.DataFrame({"Greetings": ["Greetings to you too", "hi", "hello", "hey", "greetings", "sup", "what's up", "yo"]})
df2 = pd.DataFrame({"Farewell": ["GoodBye", "See you", "Bye", "Laters"]})

frames = [df1, df2]
df3=pd.concat(frames, axis=1, join='outer', copy=True)


sentence = 'hi'
for index, row in df3.iterrows():
   if index>0:
       if sentence.lower() in df3.iloc[index,:].values:
           print(df3.iloc[index].values)

这给了我一个输出:

['hi' 'See you']

最终目的是基本遍历从第2行开始的所有列以查找关键字匹配(这就是我放置的原因:if index> 0)并且如果找到匹配,则默认输出第一项该栏目(在这种情况下,“也向你问候”,在这种情况下是df3.iloc [0,0]。

因此我需要帮助来添加额外的代码行,以便运行代码后的结果输出如下所示:

"Greetings to you too"

有人可以帮忙吗。

1 个答案:

答案 0 :(得分:0)

我终于得到了答案,代码只需要调整一下:

for index, row in df3.iterrows():
            if index>0:
                if sentence.lower() in df3.iloc[index,:].values:
                    print((df3.iloc[:index]).iloc[0,0])