我如何将一个Dataframe列一分为二，但要考虑到有时没有要拆分的内容，并且该值属于第二列？

Question

想象一下我有：

          COLUMN A
0    00000-UNITED STATES
1    01000-ALABAMA
2    01001-Autauga County, AL
3    01003-Baldwin County, AL
4    Barbour County, AL

我想将它们拆分为两列，但要确保如果最后一行中的值是字符串，它将转到第二列。如果是int，则转到第一列。例如，使用字符串：

          COLUMN B       COLUMN C
0          00000      UNITED STATES
1          01000         ALABAMA
2          01001     Autauga County, AL
3          01003     Baldwin County, AL
4                    Barbour County, AL

我试过这个：

df[['B','C']] = df.A.str.split(" - ", n = 1, expand=True)

很明显，它的反应是：

          COLUMN B       COLUMN C
0          00000      UNITED STATES
1          01000         ALABAMA
2          01001     Autauga County, AL
3          01003     Baldwin County, AL
4     Barbour County, AL

Answer 1

您可以创建两个函数，从列A中提取所需的元素并分配给列B和列C：

def get_col_b(item):
    if '-' in item:
        return item.split('-')[0]
    else:
        return ''

def get_col_c(item):
    if '-' in item:
        return item.split('-')[1]
    else:
        return item

创建列，然后删除列A：

df['COLUMN B'] = df['COLUMN A'].apply(get_col_b)
df['COLUMN C'] = df['COLUMN A'].apply(get_col_c)
cols = ['COLUMN B', 'COLUMN C']
df = df[cols]