python - Pandas Dataframe，当某些行可能有超过 1 ","时，如何将一列分成两列 ","

Question

我有一个数据框。其中一列是 CITY 和 STATE 的组合。我想将此列拆分为两列，CITY 和 STATE 使用:

df['CITY'],df['STATE'] = df['WORKSITE'].str.split(",")

我收到了这个错误:

ValueError Traceback (most recent call last) in () ----> 1 df['CITY'],df['STATE'] = df['WORKSITE'].str.split(",")

ValueError: too many values to unpack (expected 2)

那么，我想知道是否有一种方法可以忽略异常或检测哪一行不起作用？

Answer 1

在 split 调用中设置 n=2 并有效地使用 str 方法:

import pandas as pd

x = ['New York, NY', 'Los Angeles, CA', 'Kansas City, KS, US']

df = pd.DataFrame(x, columns=['WORKSITE'])

df['CITY'], df['STATE'] = df['WORKSITE'].str.split(',', 2).str[0:2].str

print df

输出

              WORKSITE         CITY STATE
0         New York, NY     New York    NY
1      Los Angeles, CA  Los Angeles    CA
2  Kansas City, KS, US  Kansas City    KS

我通过查看 this answer 得到了一些帮助至 this question .