如何根据数据框中的另一列更改一列的值

Question

我有一个数据框df。我想为所有那些数据帧的列i等于列表['S'，'T'，'U']中该值的1的行设置A列的值等于1。 我正在做类似的事情，但是没有用。它正在减小我的数据框大小。

df.loc[df['A'] == 'S', 'i'] = 1
df.loc[df['A'] == 'T', 'i'] = 1
df.loc[df['A'] == 'U', 'i'] = 1

我还想在数据框df中创建新列c，其值等于

df['c'] = df['b']*df['a'] where a is not NULL
df['c'] = df['b'] where a is NULL

有人可以建议一种有效的方法吗？