迭代 Pandas 数据框的最佳方法？

Question

我有一个这样的数据框：

x  y  someVal someOtherVal
1  2  hello   heyhey
2  1  hello   heyhey

并希望遍历每一行以检查 x

x  y  someVal someOtherVal
1  2  helloLT heyhey
2  1  hello   heyhey

我在文档中读到迭代不是好的做法，可能导致不正确的结果，所以我不确定该怎么做。

Answer 1

通常可以而且应该避免显式迭代。内部向量化操作比 Python for 循环快得多。

在这种特定情况下，使用

df.loc[df['x'] < df['y'], 'someVal'] += 'LT'

谢谢！有没有办法添加条件以仅触发小于检查 if someVal == someOtherVal？

df.loc[(df['x'] < df['y']) & (df['someVal'] == df['someOtherVal']), 'someVal'] += 'LT'

或

df.loc[df['x'].lt(df['y']) & df['someVal'].eq(df['someOtherVal']), 'someVal'] += 'LT'

Answer 2

你可以试试这个：

df['someVal'] = df['someVal'] + np.where(df['x']<df['y'],'LT','')

输出：

   x  y  someVal someOtherVal
0  1  2  helloLT       heyhey
1  2  1    hello       heyhey

pandas 将使用索引对齐在一个矢量化步骤中完成所有这些操作。