应用于 DataFrame 中特定行的不同方程

Question

我正在尝试将密度列添加到数据框中，该列由所示公式计算得出。我需要在某一行（第 66 行）更改参考密度。我尝试了两个 for 循环和一个 for/else 循环，它们每个都为每一行使用一个密度，而不是按照我的意图切换。我已经定义了所有变量，并且没有收到任何错误。

尝试 #1

for index, row in tem80df[tem80df['DREF'] <= 66].iterrows(): 
    density80 = density_crust * (1-(alpha * tem80df['TREF']))
for index, row in tem80df[tem80df['DREF'] > 66].iterrows():
    density80= density_mantle * (1- (alpha *tem80df['TREF']))

尝试 #2

for index, row in tem80df[tem80df['DREF'] <= 66].iterrows(): 
    density80 = density_crust * (1-(alpha * tem80df['TREF']))
else: 
    density80 = density_mantle * (1 - (alpha* tem80df['TREF']))

“DREF”和“TREF”是列名。 'DREF' 是第一列。

我们将不胜感激！

Answer 1

density_list = []
TR = <index of column TREF>

# get row count and loop through number of rows in dataframe
for i in range(0,tem80df.shape[0]):

    if i<=66:
        density80 = density_crust * (1-(alpha * tem80df.iloc[i:TR]))
        density_list.append(density80)

    else:
        density80= density_mantle * (1- (alpha *tem80df.iloc[i:TR]))
        density_list.append(density80)

tem80df['Density'] = density_list

TR 是列 TREF 的索引，从第一列 = 0 开始。