如何使用函数和for循环比较数据帧中的两列？

Question

我有两个柱子，共两层

df = pd.DataFrame({"floor": [1,2,30], "floors_total": [1, 50, 30]})

我想写一个函数，比如如果在一行中，floors total和floor中的两个值相等，那么返回“last floor”，如果floor=1，那么返回“first floor”，如果other返回“other” 我想把这个应用到df

我正在尝试编写一个for循环，它是：

def new(df):
  a = "floor"
  b = "floors_total"
  for a,b in df.iteritems():
    if a == b:
      return "last floor"
    elif a == 1:
      return "first floor"
    else:
      return "other"
df["floor_postiton"] = df.apply(new)

我将新列中的所有行作为NaN

Answer 1

您可以使用np.where避免数据帧上的（缓慢）迭代：

df['floor_position'] = np.where(df['floor'] == df['floors_total'],
                'last floor',
                np.where(df['floor'] == 1, 'first floor',
                     'other'))