如何根據pandas中另一列的條件添加新列？

Question

我有一個 dataframe，它包含一些變量，例如國家/地區、收入、成本，並且我有一個常數值。

我想根據我定義的某個范圍創建新列，我試過這個：

my_df['flag'] = my_df.apply(lambda row: row.cost - constant < row.revenue < row.cost + constant, 1,0)

它給了我一個結果，但標志列充滿了“假”值。 問題是什么？

Answer 1

使用Series.between和numpy.where ：

my_df['flag'] = np.where(my_df.revenue.between(my_df.cost - constant, 
                                               my_df.cost + constant,
                                               inclusive='neither'), 1, 0)

或者：

my_df['flag'] = my_df.revenue.between(my_df.cost - constant, 
                                      my_df.cost + constant,
                                      inclusive='neither').astype(int)