根据一列的条件和熊猫中另一列的值创建新列

Question

我有一个 DF，时间以小时、分钟或毫秒表示。 time 列的类型为 float，time_unit 列指示它是按小时、分钟还是毫秒给出。

我想创建一个新列来计算秒数。 因此，我需要一个函数，它首先检查它是什么 time_unit，然后从时间中获取值并将其转换为秒。

例如：

if df["time_unit"]="h":
   return df["time"]*60*60 # given hours as int
elseif: ...

我的 df 看起来像这样：

我想创建绿色列（秒）。 那么，我该如何在熊猫中做到这一点？

Answer 1

您可以创建映射dict

d = {'h' : 60*60, 'min' : 60, 'ms' : 1/1000}

df['seconds'] = df['time_unit'].map(d) * df['time']

Answer 2

seconds = []
for i,r in df.iterrows():
  if r['time_unit'] == 'h':
    seconds.append(r['time']*3600)
  elif r['time_unit'] == 'min':
    seconds.append(r['time']*60)
  elif r['time_unit'] == 'ms':
    seconds.append(r['time']/1000)
df['seconds'] = seconds

我想这会奏效。