python - 使用公式将值复制到新列

Question

我有以下DataFrame:

  geo col1  col2
0  CA    A     1
1  CA    A     2
2  CA    B     1
3  CA    B     2
4  CA    B     1
5  CA    C     1
6  CA    C     2

我的目标是创建一个新的列，如果col2 == 1，则从col2复制值，并从复制值>col1 如果col2 == 2。

图像中包含 geo 列，因为最终我想创建一个 if 语句，该语句对于美国 15-20 个州的上述操作略有不同。

最终目标如下所示:

  geo col1  col2 col3
0  CA    A     1    1
1  CA    A     2    A
2  CA    B     1    1
3  CA    B     2    B
4  CA    B     1    1
5  CA    C     1    1
6  CA    C     2    C

Answer 1

您可以将条件复制语句包装到一个函数中，以使各种情况更易于管理 (see docs) :

def copy_function(row):
    if row['col2']==1:
        return row['col2'] 
    else: 
        return row['col1']
df['col3'] = df.apply(copy_function, axis=1)