python - 如何在理解列表中迭代数据框的索引和列？

Question

如果我需要迭代一个 df在理解列表中，我会这样做:

df['new_col'] = [x if y == 1 and z == 2 for x,y in df[['col_1', 'col_2']].values]

如果不是迭代 col_1和 col_2 ，我需要迭代 df.index和 df_col_2值(value)观？
此 for 中的理解列表中的语法是什么？有条件的例子？

Answer 1

pandas 数据框和系列具有迭代方法。因此，要迭代索引和给定的列，您可以使用 iteritems:

df['new_col'] = [x if y == '1' and z =='2' for x, y in df['col_2'].iteritems()]

在这种情况下，x 是 index y 列的值 col2更一般的 iterrows使您可以在一次迭代中访问索引和所有列:

for idx, row in df.iterrows():
    print("Index", idx)
    print("col1", row.col1)
    print("col2", row.col2)
    ...