Pandas new Column с пониманием списка и оператором if со ссылкой на существующие столбцы

Question

Я пытаюсь добавить в DataFrame новый столбец со списком и оператором if следующим образом:

SD['Ln(aT) ANALYTIC'] = [x + 1 for x in SD['T'] if SD['T'] >= SD['TG']]

и я получаю эту ошибку:

 The truth value of a Series is ambiguous. Use a.empty, 
 a.bool(), a.item(), a.any() or a.all().

Я не знаю, как справиться с этой проблемой.

Есть предложения?

РЕДАКТИРОВАТЬ: DataFrame выглядит как:

Answer 1

Использовать numpy.where с логической маской:

mask = SD['T'] >= SD['TG']
SD['Ln(aT) ANALYTIC'] = np.where(mask, SD['T'] + 1, SD['T'])

Или:

SD['Ln(aT) ANALYTIC'] = np.where(mask, SD['T'] + 1, np.nan)

Возможно составление списка, но медленно:

SD['Ln(aT) ANALYTIC1'] = [i + 1 if i >= j else i for i, j in zip(SD['T'], SD['TG'])]

---

SD = pd.DataFrame({'T': [1,2,3],
                   'TG':[2,5,1]})

#[3000 rows x 2 columns]
SD = pd.concat([SD] * 1000, ignore_index=True)


In [294]: %timeit SD['Ln(aT) ANALYTIC1'] = [i + 1 if i >= j else i for i, j in zip(SD['T'], SD['TG'])]
1.18 ms ± 82.6 µs per loop (mean ± std. dev. of 7 runs, 1000 loops each)

In [295]: %timeit SD['Ln(aT) ANALYTIC2'] = np.where(SD['T'] >= SD['TG'], SD['T'] + 1, SD['T'])
511 µs ± 16.8 µs per loop (mean ± std. dev. of 7 runs, 1000 loops each)