Question

比方说，A栏是基于时间的，B栏是薪水。

我在for循环中使用了一个if语句，试图查找“所有比前一个薪金少，但也比下一个薪金高的薪水”。然后，将新值（'YES'）分配给满足条件的行的另一列（C列）。最后，我想抓住满足以上条件的第一列A。

数据框如下所示：

In [1]:
df = pd.DataFrame({'A':['2007q3','2007q4','2008q1','2008q2','2008q3','2008q4','2009q1','2009q2','2009q3'], 
                   'B':[14938, 14991, 14899, 14963, 14891, 14577, 14375, 14355, 14402]})
df['C'] = pd.Series()
df

Out [1]:
      A       B      C
0   2007q3  14938   NaN
1   2007q4  14991   NaN
2   2008q1  14899   NaN
3   2008q2  14963   NaN
4   2008q3  14891   NaN
5   2008q4  14577   NaN
6   2009q1  14375   NaN
7   2009q2  14355   NaN
8   2009q3  14402   NaN

下面的代码可以正常工作，但是显示“ SettingWithCopyWarning”警告，我不确定代码的哪一部分会导致问题...

In [2]:
for i in range(1, len(df)-1):
    if (df['B'][i] < df['B'][i-1]) & (df['B'][i] > df['B'][i+1]):
       df['C'][i] = 'YES'
df

Out [2]: 
      A       B      C
0   2007q3  14938   NaN
1   2007q4  14991   NaN
2   2008q1  14899   NaN
3   2008q2  14963   NaN
4   2008q3  14891   YES
5   2008q4  14577   YES
6   2009q1  14375   YES
7   2009q2  14355   NaN
8   2009q3  14402   NaN

In [3]: df['A'][df['C'] == 'YES'].iloc[0]
Out [3]:'2008q3'

或者也许有更好的方法来完成这项工作。谢谢！！！

Answer 1

有关为何获得 SettingWithCopyWarning 的更多详细信息，建议您阅读this answer。这主要是因为选择列df['C']然后选择带有[i]的行会执行“链接分配” ，当您执行df['C'][i] = 'YES' < / p>

对于尝试执行的操作，可以在B列上使用np.where和shift，例如：

import numpy as np
df['C'] = np.where((df.B < df.B.shift()) & (df.B > df.B.shift(-1)), 'YES', np.nan)

您将获得相同的输出。

如何在for循环if语句中处理“ SettingWithCopyWarning”

1 个答案: