基于另一个列的值，新列中的 Pandas Max 或 Min Row 值

Question

所以熊猫数据框中有 3 列我想创建第 4 列 D，这将基于这样一个条件，即如果 A > 5 那么 D 的值为 B，否则值为 C。这是针对每一行的A,B,C    1,1,26,2,37,3,1例如，如果您看到新列 DD row1 = 1 # 因为 A < 5，所以会选择 B 行D row2 = 3 # 因为 A > 5，所以会选择 C 行D row3 = 1 # 因为 A > 5，所以会选择 C 行所以最终的数据帧会像A,B,C,D    1,1,2,16,2,3,37,3,1,1

Answer 1

我想你可以用

import numpy as np
df['D'] = np.where(df['A']>5, df['C'], df['B'])