如何使用python循环迭代函数中的值?

时间:2018-01-18 16:24:29

标签: python pandas loops automation

我想使用python在函数中逐个循环传递值。值存储在数据帧中。

def eam(A,B):
    y=A +" " +B
    return y

假设我将A的值作为国家/地区传递B作为资本。 数据框df

country                   capital
India                     New Delhi
Indonesia                 Jakarta
Islamic Republic of Iran  Tehran
Iraq                      Baghdad
Ireland                   Dublin

如何使用循环

获取价值
0   India New Delhi
1   Indonesia Jakarta
2   Islamic Republic of Iran Tehran
3   Iraq Baghdad
4   Ireland Dublin

1 个答案:

答案 0 :(得分:1)

在这里,您只需使用以下语法在数据框中获取新列。无需编写代码来遍历行。但是,如果必须循环,则df.iterrows()返回或df.itertuples()提供良好的功能来实现类似的目标。

>>> df = pd.read_clipboard(sep='\t')
>>> df.head()
                    country    capital
0                     India  New Delhi
1                 Indonesia    Jakarta
2  Islamic Republic of Iran     Tehran
3                      Iraq    Baghdad
4                   Ireland     Dublin
>>> df.columns
Index(['country', 'capital'], dtype='object')
>>> df['both'] = df['country'] + " " + df['capital']
>>> df.head()
                    country    capital                             both
0                     India  New Delhi                  India New Delhi
1                 Indonesia    Jakarta                Indonesia Jakarta
2  Islamic Republic of Iran     Tehran  Islamic Republic of Iran Tehran
3                      Iraq    Baghdad                     Iraq Baghdad
4                   Ireland     Dublin                   Ireland Dublin