擅长:python、mysql、java
<p>可以使用列表理解和if-else逻辑创建列</p>
<pre><code># Sample data
df = pd.DataFrame({'TGR1':[random.randint(0,3) for i in range(10)],
'1':[random.randint(0,100) for i in range(10)],
'2':[random.randint(101,200) for i in range(10)],
'3':[random.randint(201,300) for i in range(10)]})
# creating the column
df['BB'] = [0 if tgr1_val == 0 else df.loc[ind,str(tgr1_val)]
for ind,tgr1_val in enumerate(df['TGR1'].values)]
df
# TGR1 1 2 3 BB
# 0 0 54 107 217 0
# 1 2 71 128 277 128
# 2 1 25 103 269 25
# 3 0 80 112 279 0
# 4 2 98 167 228 167
# 5 3 26 192 285 285
# 6 0 27 107 228 0
# 7 2 13 103 298 103
# 8 3 28 196 289 289
# 9 2 72 186 251 186
</code></pre>